| 411 | } |
| 412 | |
| 413 | std::unique_ptr<fft::Plan> ROCMFft::Create2dPlan(Stream *stream, uint64 num_x, |
| 414 | uint64 num_y, fft::Type type, |
| 415 | bool in_place_fft) { |
| 416 | std::unique_ptr<ROCMFftPlan> fft_plan_ptr{new ROCMFftPlan()}; |
| 417 | uint64 elem_count[2] = {num_x, num_y}; |
| 418 | port::Status status = fft_plan_ptr->Initialize( |
| 419 | parent_, stream, 1, elem_count, type, /*scratch_allocator=*/nullptr); |
| 420 | if (!status.ok()) { |
| 421 | LOG(FATAL) << "failed to initialize hipfft 2d plan: " |
| 422 | << status.error_message(); |
| 423 | } |
| 424 | return std::move(fft_plan_ptr); |
| 425 | } |
| 426 | |
| 427 | std::unique_ptr<fft::Plan> ROCMFft::Create2dPlanWithScratchAllocator( |
| 428 | Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, |
nothing calls this directly
no test coverage detected