mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-11-02 08:33:19 +08:00
Import layout oracle from the nvfp4 submodule in the language test
Follow-up to the package export slimming: blockscaled_chunk_kmajor_word_offset is no longer re-exported at the tilelang.quantize package level. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
@@ -7,7 +7,7 @@ from tilelang import tvm
|
||||
from tilelang.cuda.intrinsics.layout.mma_layout import mma_load_a_32x32_to_shared_16x64_layout
|
||||
from tilelang.cuda.intrinsics.macro.mma_macro_generator import SM120BlockScaledFullTilePackageContract
|
||||
from tilelang.intrinsics import TensorCoreIntrinEmitter, get_swizzle_layout
|
||||
from tilelang.quantize import blockscaled_chunk_kmajor_word_offset
|
||||
from tilelang.quantize.nvfp4 import blockscaled_chunk_kmajor_word_offset
|
||||
from tilelang.transform import simplify_prim_func
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user