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

Function gemm_default_solution

src/targets/gpu/gemm_impl.cpp:597–605  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

595#endif
596
597int32_t gemm_default_solution(context& ctx,
598 const shape& output_shape,
599 const std::vector<shape>& input_shapes)
600{
601 auto sol = ctx.get_problem_cache().get("rocblas", gemm_problem(output_shape, input_shapes));
602 if(sol.has_value())
603 return sol->to<int32_t>();
604 return 0;
605}
606
607/**
608 * Decides if the tune() or validate() method is appropriate and calls it.

Callers 1

finalizeMethod · 0.85

Calls 2

gemm_problemFunction · 0.85
getMethod · 0.45

Tested by

no test coverage detected