tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix?

Browse Source
This commit is contained in:
Dan Saunders
2025-09-23 16:21:00 -04:00
parent d0da67eb17
commit a81612305c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/kernels/moe/tt_mg_gemm/tma_autotuning.py
View File

@@ -77,8 +77,8 @@ class TmaDescriptorHelper:
) )
self.tma_size = tma_size self.tma_size = tma_size
self.fill_1d_tma_descriptor_inner = driver.active.utils.fill_1d_tma_descriptor self.fill_1d_tma_descriptor_inner = driver.active.utils.fill_tma_descriptor
self.fill_2d_tma_descriptor_inner = driver.active.utils.fill_2d_tma_descriptor self.fill_2d_tma_descriptor_inner = driver.active.utils.fill_tma_descriptor
self.descriptors: Dict[str, torch.Tensor] = {} self.descriptors: Dict[str, torch.Tensor] = {}
def init_tma_descriptor(self, name: str) -> None: def init_tma_descriptor(self, name: str) -> None: