| 470 | } |
| 471 | |
| 472 | std::unique_ptr<fft::Plan> ROCMFft::CreateBatchedPlan( |
| 473 | Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, |
| 474 | uint64 input_stride, uint64 input_distance, uint64 *output_embed, |
| 475 | uint64 output_stride, uint64 output_distance, fft::Type type, |
| 476 | bool in_place_fft, int batch_count) { |
| 477 | std::unique_ptr<ROCMFftPlan> fft_plan_ptr{new ROCMFftPlan()}; |
| 478 | port::Status status = fft_plan_ptr->Initialize( |
| 479 | parent_, stream, rank, elem_count, input_embed, input_stride, |
| 480 | input_distance, output_embed, output_stride, output_distance, type, |
| 481 | batch_count, /*scratch_allocator=*/nullptr); |
| 482 | if (!status.ok()) { |
| 483 | LOG(FATAL) << "failed to initialize batched hipfft plan: " |
| 484 | << status.error_message(); |
| 485 | } |
| 486 | |
| 487 | return std::move(fft_plan_ptr); |
| 488 | } |
| 489 | |
| 490 | std::unique_ptr<fft::Plan> ROCMFft::CreateBatchedPlanWithScratchAllocator( |
| 491 | Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, |
nothing calls this directly
no test coverage detected