Cong Ma 5abe4109e0 Introduces the new partitioner to implement the reduction StreamK kernel. (#3107) ...

* Introduces the new partitioner to implement the reduction StreamK kernel

* Add more doc text to functions

* Add persistent-dp option to streamk example

* Update example/ck_tile/40_streamk_gemm/README.md

2025-11-04 10:32:17 -07:00

..

reference

[CK_TILE] B matrix 2D block scale gemm (#3074)

2025-11-02 16:49:20 -08:00

arg_parser.hpp

Apply Ck-tile argument parser for vectors [I/O] (#1758)

2024-12-19 17:55:35 +01:00

check_err.hpp

[CK_Tile] Merge multiple convolution groups into a single GEMM batch (#2986)

2025-10-29 16:49:28 +02:00

concat.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

convolution_host_tensor_descriptor_helper.hpp

Simplify static_cast if-lands (#1828)

2025-01-21 23:23:19 +01:00

convolution_parameter.hpp

[CK_TILE] Fix conv param multiple definition (#1550)

2024-10-07 15:21:21 +02:00

device_memory.hpp

Add Doxygen Documentation for HostTesnor, HostTensorDescriptor, DeviceMem, FillUniformDistribution (#2160)

2025-05-21 10:34:30 -07:00

device_prop.hpp

[CK_TILE] CK_TILE GEMM WMMA Support for GFX11/GFX12 (#2466)

2025-08-15 16:22:27 -07:00

fill.hpp

[CK_TILE] Refine pk_fp4's fill, pack, and unpack (#2845)

2025-09-17 10:54:06 +08:00

flush_icache.hpp

docs: add inline comments about flush_cache and rotating buffer

2025-10-17 12:56:47 -04:00

hip_check_error.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

host_tensor.hpp

feat(HostTensor): Extend support for HostTensor class' >> operator to print more data types (#2691)

2025-08-27 18:17:24 -07:00

joinable_thread.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

kernel_launch.hpp

Introduces the new partitioner to implement the reduction StreamK kernel. (#3107)

2025-11-04 10:32:17 -07:00

permute_pk_int4.hpp

[CK TILE] Support fp8/fp16 with pk_int4_t as data types for tensors A and B (#2805)

2025-09-09 16:40:52 -07:00

ranges.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

rotating_buffers.hpp

limit the rotating count to prevent oom (#3087)

2025-10-24 08:55:34 -07:00

stream_config.hpp

[CK_Tile] Updating gpu timer when doing flush cache (#2593)

2025-07-31 16:43:33 -07:00

stream_utils.hpp

[CK_TILE] Grouped GEMM tile loop (#2146)

2025-05-20 17:18:57 +03:00

tensor_shuffle_utils.hpp

WMMA gemm_add_relu_add_layernorm (#2989)

2025-10-31 11:19:26 -07:00

timer.hpp

[CK_Tile] Updating gpu timer when doing flush cache (#2593)

2025-07-31 16:43:33 -07:00