Mojo module
block_scaled_tile_pipeline
Block-scaled tile pipeline for SM100 matmul.
Extends TilePipeline to manage 4 tile arrays (A, B, SFA, SFB) with synchronized producer-consumer access for TMA loading and MMA consumption.
comptime values
MbarPtr
comptime MbarPtr = LegacyUnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED]
Structs
-
BlockScaledConsumerStage: Context manager for consumer tile access with scaling factors. -
BlockScaledInputConsumer: Consumer view for MMA warp (block-scaled input pipeline). -
BlockScaledInputProducer: Producer view for TMA Load warp (block-scaled input pipeline). -
BlockScaledProducerStage: Context manager for producer tile access with scaling factors. -
BlockScaledTilePipeline: Staged tile storage for A, B, SFA, SFB with producer-consumer sync.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!