From d41912cd7fcac81920658f226d4a7fcbfe530836 Mon Sep 17 00:00:00 2001 From: Matthew Noto Date: Wed, 12 Nov 2025 13:34:00 -0800 Subject: [PATCH] fix import --- nvfp4_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nvfp4_utils.py b/nvfp4_utils.py index b95012e9c..244b372fb 100644 --- a/nvfp4_utils.py +++ b/nvfp4_utils.py @@ -4,7 +4,7 @@ import triton import triton.language as tl -from triton_kernels.target_info import cuda_capability_geq +from triton.language.target_info import cuda_capability_geq MXFP_BLOCK_SIZE = tl.constexpr(16)