1. 23 3月, 2022 1 次提交
    • K
      Add complex type compatibility for stft api and stft op. (#40113) · 319f95d0
      KP 提交于
      * Add stft_op.
      
      * Add stft_grad_op.
      
      * Add stft_op unittest.
      
      * [DLTP-45176] Add complex compatibility in static mode for stft api.
      
      * [DLTP-45176] Add complex compatibility in static mode for stft api.
      
      * Add doc.
      
      * Update unitests of stft op.
      
      * Update spectral helper.
      
      * fix coding style.
      319f95d0
  2. 03 12月, 2021 1 次提交
  3. 27 10月, 2021 1 次提交
    • F
      Add LRUCache for fft plans (#36646) · 737992eb
      Feiyu Chan 提交于
      * WIP: add cache
      
      * delete move constructor and operator= for CuFFTHandle and FFTConfig
      
      * remove log from CuFFTHandle and FFTConfig
      
      * add lrucache for fft rocm backend
      
      * disable LRUCache when CUFFT_VERSION >= 10200
      
      * disbale copy and move for hipFFTHandle; format code
      
      * clean debug code
      Co-authored-by: NXiaoxu Chen <chenxx_id@163.com>
      737992eb
  4. 19 10月, 2021 1 次提交