OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ROCm][TunableOp] Remove extra transpose characters in hipBLASLt signature. (#147900)

Browse Source 
Cleanup the TunableOp hipBLASLt signature of extra transpose characters.

Test manually and no new regressions found.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147900
Approved by: https://github.com/jeffdaily
This commit is contained in:
Nichols A. Romero 2025-02-26 22:27:57 +00:00 committed by PyTorch MergeBot
parent 7e7d05bf85
commit b13ad1a193
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/cuda/tunable/GemmHipblaslt.h
View File

@ -631,7 +631,7 @@ auto GetHipBlasLtTypeStringAndOps() {
auto algo = heuristic_result[i].algo; auto algo = heuristic_result[i].algo;
int algo_index = hipblaslt_ext::getIndexFromAlgo(algo); int algo_index = hipblaslt_ext::getIndexFromAlgo(algo);
auto callable = std::make_unique<HipblasltGemmOp<AT, BT, CT, ALayout, BLayout, ParamsT>>(algo); auto callable = std::make_unique<HipblasltGemmOp<AT, BT, CT, ALayout, BLayout, ParamsT>>(algo);
std::string type_string = fmt::sprintf("Gemm_Hipblaslt_%c%c_%d", _charFromhipblasOp(transa_outer), _charFromhipblasOp(transb_outer), algo_index); std::string type_string = fmt::sprintf("Gemm_Hipblaslt_%d", algo_index);
ret.emplace_back(type_string, std::move(callable)); ret.emplace_back(type_string, std::move(callable));
} }