transformer_engine/common/gemm/rocm_gemm.cu · 388ac7354d4704ae6340e2443040b6cedfc265f4 · OpenDAS / TransformerEngine

wenjh authored Apr 30, 2025


Signed-off-by: wenjh <wenjh@sugon.com>

[RocblasGemm] Provide support of AB(bf16)D(fp32)
Signed-off-by: wenjh <wenjh@sugon.com>

388ac735

rocm_gemm.cu 73 KB