Mojo module
utils
comptime values
LocalLayoutTensor
comptime LocalLayoutTensor[dtype: DType, layout: Layout] = LayoutTensor[dtype, layout, MutAnyOrigin, address_space=AddressSpace.LOCAL]
Parameters
SharedLayoutTensor
comptime SharedLayoutTensor[dtype: DType, layout: Layout] = LayoutTensor[dtype, layout, MutAnyOrigin, address_space=AddressSpace.SHARED]
Parameters
Structs
Functions
-
copy_dram_to_sram_lds: -
copy_local_to_dram2: -
get_fragment_layout: -
get_nested_fragment_layout: -
get_warp_coords: -
get_warp_layout: -
load_b: -
load_b_: -
load_b_tr: Loads the b operand tile for AMD tensor core MFMA instructions using transposed memory access. -
pad:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!