| 440 | } |
| 441 | |
| 442 | std::unique_ptr<fft::Plan> ROCMFft::Create3dPlan(Stream *stream, uint64 num_x, |
| 443 | uint64 num_y, uint64 num_z, |
| 444 | fft::Type type, |
| 445 | bool in_place_fft) { |
| 446 | std::unique_ptr<ROCMFftPlan> fft_plan_ptr{new ROCMFftPlan()}; |
| 447 | uint64 elem_count[3] = {num_x, num_y, num_z}; |
| 448 | port::Status status = fft_plan_ptr->Initialize( |
| 449 | parent_, stream, 3, elem_count, type, /*scratch_allocator=*/nullptr); |
| 450 | if (!status.ok()) { |
| 451 | LOG(FATAL) << "failed to initialize hipfft 3d plan: " |
| 452 | << status.error_message(); |
| 453 | } |
| 454 | return std::move(fft_plan_ptr); |
| 455 | } |
| 456 | |
| 457 | std::unique_ptr<fft::Plan> ROCMFft::Create3dPlanWithScratchAllocator( |
| 458 | Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, |
nothing calls this directly
no test coverage detected