| 267 | } |
| 268 | |
| 269 | int CUDAFftPlan::GetFftDirection() const { |
| 270 | if (!IsInitialized()) { |
| 271 | LOG(FATAL) << "Try to get fft direction before initialization."; |
| 272 | } else { |
| 273 | switch (fft_type_) { |
| 274 | case fft::Type::kC2CForward: |
| 275 | case fft::Type::kZ2ZForward: |
| 276 | case fft::Type::kR2C: |
| 277 | case fft::Type::kD2Z: |
| 278 | return CUFFT_FORWARD; |
| 279 | case fft::Type::kC2CInverse: |
| 280 | case fft::Type::kZ2ZInverse: |
| 281 | case fft::Type::kC2R: |
| 282 | case fft::Type::kZ2D: |
| 283 | return CUFFT_INVERSE; |
| 284 | default: |
| 285 | LOG(FATAL) << "Invalid value of fft::Type."; |
| 286 | } |
| 287 | } |
| 288 | } |
| 289 | |
| 290 | std::unique_ptr<fft::Plan> CUDAFft::Create1dPlan(Stream *stream, uint64 num_x, |
| 291 | fft::Type type, |
no outgoing calls
no test coverage detected