| 56 | // Get FFT direction in cuFFT based on FFT type. |
| 57 | int GetFftDirection() const; |
| 58 | cufftHandle GetPlan() const { |
| 59 | if (IsInitialized()) { |
| 60 | return plan_; |
| 61 | } else { |
| 62 | LOG(FATAL) << "Try to get cufftHandle value before initialization."; |
| 63 | } |
| 64 | } |
| 65 | |
| 66 | // Initialize function for batched plan |
| 67 | port::Status Initialize(GpuExecutor* parent, Stream* stream, int rank, |
no outgoing calls
no test coverage detected