cleanup

Signed-off-by: Mayank Mishra <[email protected]>
mayank31398 · Jan 24, 2025 · be5cfae · be5cfae
1 parent bde5258
commit be5cfae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cute_kernels/kernels/gemm/__init__.py b/cute_kernels/kernels/gemm/__init__.py
@@ -75,8 +75,6 @@ def gemm_cute(
                 BLOCK_SIZE=BLOCK_SIZE_M,
             )
         elif cuda_kernel_algorithm == CUDAKernelAlgorithm.naive:
-            assert cuda_kernel_algorithm is None
-
             naive_gemm_cuda(
                 a=a,
                 b=b,
@@ -93,6 +91,8 @@ def gemm_cute(
                 BLOCK_SIZE_N=BLOCK_SIZE_N,
             )
     elif kernel_backend == KernelBackend.triton:
+        assert cuda_kernel_algorithm is None
+
         gemm_triton(
             a=a,
             b=b,