biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix test: use from_dlpack for c_tensor

Browse Source
This commit is contained in:
biondizzle
2026-06-01 07:55:29 +00:00
parent d2819fc39c
commit 25b9a5f32d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit/test_nvfp4_cutedsl_compile.py
View File

@@ -54,8 +54,10 @@ def test_nvfp4_cutedsl_compilation():
sfb_tensor = cutlass_torch.from_dlpack(scale_b)
sfb_tensor = sfb_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim(scale_b))
c_tensor = cutlass_torch.as_tensor(
c_tensor = cutlass_torch.from_dlpack(
torch.empty(M, N, dtype=torch.bfloat16, device=device))
c_tensor = c_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim(
torch.empty(M, N, dtype=torch.bfloat16, device=device)))
print("CuTe tensors created OK")