mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-05 22:22:27 +00:00
Add support for full types (not just aliases) in type-print
- Added support for static_distributed_tensor<...> - Added support for tile_distribution<...> - Added support for tensor_view<...> - Added support for tensor_descriptor<...> Now type-print handles both: 1. Type aliases (::BottomTensorView, ::TensorDesc, etc.) 2. Full types with no runtime storage (static_distributed_tensor, etc.) Shows [from type] indicator for all type-only extractions. Example: type-print dst_tensor Works even when 'p dst_tensor' shows 'Cannot access memory' Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -103,7 +103,7 @@ int run_flatmm_example_with_layouts(int argc,
|
||||
}
|
||||
else
|
||||
{
|
||||
return shuffle_b<FlatmmConfig>(b_origin_host);
|
||||
return moe_shuffle_b<FlatmmConfig>(b_origin_host);
|
||||
}
|
||||
}();
|
||||
ck_tile::DeviceMem b_shuffle_dev_buf(b_shuffle_host.get_element_space_size_in_bytes());
|
||||
|
||||
Reference in New Issue
Block a user