MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / gemm_save_solution

Function gemm_save_solution

src/targets/gpu/gemm_impl.cpp:587–594  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

585
586#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
587static void gemm_save_solution(context& ctx,
588 const shape& output_shape,
589 const std::vector<shape>& input_shapes,
590 int32_t solution_idx)
591{
592 ctx.get_problem_cache().insert(
593 "rocblas", gemm_problem(output_shape, input_shapes), solution_idx);
594}
595#endif
596
597int32_t gemm_default_solution(context& ctx,

Callers 1

gemm_finalize_implFunction · 0.85

Calls 2

gemm_problemFunction · 0.85
insertMethod · 0.45

Tested by

no test coverage detected