working mkl alloc

hysop/numerics/fft/_mkl_fft.py

@@ -19,16 +19,16 @@ from mkl_fft import (ifft as mkl_ifft,
                      rfft_numpy as mkl_rfft, 
                      irfft_numpy as mkl_irfft)
 
-from hysop.tools.types import first_not_None
 from hysop.numerics.fft.host_fft import HostFFTPlanI, HostFFTI, HostArray
 from hysop.numerics.fft.fft import \
             complex_to_float_dtype, float_to_complex_dtype, \
             mk_view, mk_shape, simd_alignment
 
 from hysop import __DEFAULT_NUMBA_TARGET__
-from hysop.tools.numba_utils import make_numba_signature, prange
 from hysop.numerics.fft.fft import HysopFFTWarning, bytes2str
+from hysop.tools.numba_utils import make_numba_signature, prange
 from hysop.tools.warning import HysopWarning
+from hysop.tools.types import get_itemsize
 
 class HysopMKLFftWarning(HysopWarning):
     pass
@@ -433,13 +433,13 @@ class MklFFTPlan(HostFFTPlanI):
         alignment = simd_alignment
         if self._required_input_tmp:
             sin, din   = self._required_input_tmp['size'], self._required_input_tmp['dtype']
-            sin *= din.itemsize
+            sin *= get_itemsize(din)
             Bin  = ((sin+alignment-1)//alignment)*alignment
         else:
             Bin = 0
         if self._required_output_tmp:
             sout, dout = self._required_output_tmp['size'], self._required_output_tmp['dtype']
-            sout *= dout.itemsize
+            sout *= get_itemsize(dout)
             Bout = ((sout+alignment-1)//alignment)*alignment
         else: 
             Bout = 0
@@ -455,13 +455,13 @@ class MklFFTPlan(HostFFTPlanI):
             alignment = simd_alignment
             if self._required_input_tmp:
                 sin, din   = self._required_input_tmp['size'], self._required_input_tmp['dtype']
-                sin *= din.itemsize
+                sin *= get_itemsize(din)
                 Bin  = ((sin+alignment-1)//alignment)*alignment
             else:
                 Bin = 0
             if self._required_output_tmp:
                 sout, dout = self._required_output_tmp['size'], self._required_output_tmp['dtype']
-                sout *= dout.itemsize
+                sout *= get_itemsize(dout)
                 Bout = ((sout+alignment-1)//alignment)*alignment
             else: 
                 Bout = 0