Fix the CuDNN failure on bmm_fp8 when two libcudart.so exists. (#29201)

Co-authored-by: Brayden Zhong <brayden@radixark.ai>
2026-06-24 15:26:46 -07:00
parent 5f30fa258c
commit 3c95a87b66
1 changed files with 1 additions and 1 deletions
@@ -270,7 +270,7 @@ if is_blackwell_supported() and is_flashinfer_available():
            x_scale.reshape(1),
            weight_scale.reshape(1),
            out_dtype,
-            backend="auto",
+            backend="cublas",
        ).view(m, n)

    @lru_cache(maxsize=1)