diff --git a/src/axolotl/kernels/moe/tt_mg_gemm/tma_autotuning.py b/src/axolotl/kernels/moe/tt_mg_gemm/tma_autotuning.py index f77ecbfeb..53b1c333b 100644 --- a/src/axolotl/kernels/moe/tt_mg_gemm/tma_autotuning.py +++ b/src/axolotl/kernels/moe/tt_mg_gemm/tma_autotuning.py @@ -15,7 +15,6 @@ from typing import Dict import torch import triton -import triton.language as tl from triton.runtime import driver # @manual sys.path.append(os.path.dirname(os.path.abspath(__file__))) @@ -71,10 +70,6 @@ class TmaDescriptorHelper: raise RuntimeError( "TMA not supported on this device (requires Hopper or newer)" ) - if "nv_tma_desc_type" not in dir(tl): - raise RuntimeError( - "TMA grid constant descriptors not supported in your Triton version" - ) self.tma_size = tma_size self.fill_1d_tma_descriptor_inner = driver.active.utils.fill_tma_descriptor