Skip to content

Commit ce412ff

Browse files
author
Veera Gopu
committed
updated ci script to run new gemm tests
1 parent 2b5fa5c commit ce412ff

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

ci/core.sh

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,7 @@ for _gemm in hipblaslt rocblas; do
2626
configure_gemm_env $_gemm || continue
2727
_exclude=""
2828
if [ $_gemm = "hipblaslt" ]; then
29-
_exclude="-E Test(.*bf16/.*X.X1|.*fp8.*fp16/.*X1X0|.*fp8.*X.X1|.*fp8/|.*bf8/)"
30-
echo ===== Run FP8 GEMM gelu fusion $_gemm tests =====
31-
# fp8 gelu fusion tests
32-
ctest --test-dir build -j4 -R "OperatorTest/GEMMTestSuite.Testfp8xfp8xfp16xfp16xfp8/.*X.X1"
29+
_exclude="-E Test(.*bf16/.*X.X1|.*fp8.*fp16/.*X1X0|.*fp8xbf.*X.X1|.*bf16xfp8/|.*bf8/)"
3330
fi
3431
echo ===== Run GEMM $_gemm tests =====
3532
ctest --test-dir build -j4 -R "OperatorTest/GEMMTestSuite" $_exclude

tests/cpp/operator/test_cublaslt_gemm.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ void compute_ref(
8787
for(size_t kk = 0; kk < k; kk++){
8888
float a_val = transa ? (float)a_data[kk + ii*k] : (float)a_data[ii + kk*m];
8989
float b_val = transb ? (float)b_data[jj + kk*n] : (float)b_data[kk + jj*k];
90-
val += a_scale_inv*b_scale_inv*a_val*b_val;
90+
val += a_scale_inv*a_val*b_scale_inv*b_val;
9191
}
9292
if(bias_data){
9393
val += (float)bias_data[ii];
@@ -124,10 +124,10 @@ void performTest(bool use_bias, bool use_gelu, const size_t m, const size_t k, c
124124
}
125125
Tensor B;
126126
if (transb){
127-
B = Tensor({ k, n }, atype);
127+
B = Tensor({ k, n }, btype);
128128
}
129129
else {
130-
B = Tensor({ n, k }, atype);
130+
B = Tensor({ n, k }, btype);
131131
}
132132
Tensor D({ n, m }, dtype);
133133
Tensor bias;

0 commit comments

Comments
 (0)