onnxruntime
3940ef20 - [ROCm] Refactor to hide ck layout (Row/Col) from ORT interface (#18777)

Commit
2 years ago
[ROCm] Refactor to hide ck layout (Row/Col) from ORT interface (#18777) Previously, we use `ck::tensor_layout::gemm::RowMajor` or `ColumnMajor` to tag the template for correct dispatch. This is cumbersome in the case of CK is disabled. Switch to use the ORT BlasOp to tag the template and use `CKBlasOpAdaptor` to adapt between ORT BlasOp enum and ck's Col/Row. Just like what we have done for ORT datatype and ck datatype with `CKDataTypeAdaptor`.
Author
Parents
Loading