mindscience.sciops.fft.set_fft_cache_size

mindscience.sciops.fft.set_fft_cache_size(cache_size)[源代码]

设置ASD FFT算子的缓存数量以优化函数调用性能。 如果没有缓存,每次ASD FFT函数调用都会从.so文件中动态dlopen函数符号, 这会引入一些开销。 使用缓存后,函数符号将在第一次ASD FFT函数调用时加载到缓存中, 在后续调用中将不再重新加载。

参数:
  • cache_size (int) - ASD FFT算子的缓存数量。

样例:

>>> from mindscience.sciops.fft import set_fft_cache_size
>>> set_fft_cache_size(1024)  # Set 1024 fft operators cache number