Restore math mode

matthewdcong · matthewdcong · commit 7b60ceeb4ec9 · 2026-01-21T20:49:43.000-08:00
Signed-off-by: Matthew Cong &lt;mcong@nvidia.com&gt;
diff --git a/src/fvdb/detail/ops/convolution/backend/MESparseConvolution.cu b/src/fvdb/detail/ops/convolution/backend/MESparseConvolution.cu
@@ -128,6 +128,7 @@ gpu_gemm<at::Half>(const cublasOperation_t transA,
                               N,
                               CUDA_R_32F,
                               CUBLAS_GEMM_DEFAULT_TENSOR_OP));
+    CUBLAS_CHECK(cublasSetMathMode(handle, CUBLAS_DEFAULT_MATH));
 }
 
 template <>

Original file line number	Diff line number	Diff line change
`@@ -128,6 +128,7 @@ gpu_gemm<at::Half>(const cublasOperation_t transA,`
`128`	`128`	`N,`
`129`	`129`	`CUDA_R_32F,`
`130`	`130`	`CUBLAS_GEMM_DEFAULT_TENSOR_OP));`
	`131`	`+ CUBLAS_CHECK(cublasSetMathMode(handle, CUBLAS_DEFAULT_MATH));`
`131`	`132`	`}`
`132`	`133`
`133`	`134`	`template <>`