Fix _symm for SB_ENABLE_JOINT_MATRIX=1

codeplaysoftware · Oct 20, 2023 · eae0f8b · eae0f8b
1 parent 7e85bf9
commit eae0f8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/interface/blas3/backend/nvidia_gpu.hpp b/src/interface/blas3/backend/nvidia_gpu.hpp
@@ -55,7 +55,7 @@ typename sb_handle_t::event_t _gemm(
 
 #ifdef SB_ENABLE_JOINT_MATRIX
   const char* en_joint_matrix = std::getenv("SB_ENABLE_JOINT_MATRIX");
-  if (en_joint_matrix != NULL && *en_joint_matrix == '1') {
+  if (en_joint_matrix != NULL && *en_joint_matrix == '1' && !s_a && !s_b) {
     if (_M > 1024 && _N > 1024) {
       return blas::Gemm_Launcher<
           container_0_t, container_1_t, container_2_t, 256, false, true, true,