ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-14 11:07:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Partial Progress : Jenkins change to run op_new

Browse Source
This commit is contained in:
ThruptiRajLakshmanaGowda
2025-11-25 23:46:00 +00:00
parent d78098498c
commit 1a07c85301
3 changed files with 8 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
Jenkinsfile vendored
View File

@@ -1629,15 +1629,15 @@ pipeline {
-D CMAKE_CXX_COMPILER="${build_compiler()}" \
-D CMAKE_BUILD_TYPE=Release \
-D GPU_TARGETS="gfx90a" \
-D GEMM_DATATYPE="fp8;fp16" \
-D GEMM_LAYOUT="rcr;rrr;crr;ccr" \
-D GEMM_UNIVERSAL_DATATYPE="fp8;fp16" \
-D GEMM_UNIVERSAL_LAYOUT="rcr;rrr;crr;ccr" \
-D GEMM_MULTI_D_DATATYPE="fp16" \
-D GEMM_MULTI_D_LAYOUT="rcrr;rrrr;crrr;ccrr" \
-D GEMM_PRESHUFFLE_DATATYPE="fp16;fp8;bf16;bf8" \
-D GEMM_PRESHUFFLE_LAYOUT="rcr" \
-DCMAKE_CXX_FLAGS=" -O3 " .. && \
ninja -j64 benchmark_gemm_all && \
python3 ../tile_engine/ops/gemm/gemm_benchmark.py . --problem-sizes "1024,1024,1024" \
ninja -j64 benchmark_gemm_universal_all && \
python3 ../tile_engine/ops_new/gemm/gemm_universal/gemm_benchmark.py . --problem-sizes "1024,1024,1024" \
--warmup 5 --repeat 5 --verbose --json results.json && \
ninja -j64 benchmark_gemm_preshuffle_all && \
python3 ../tile_engine/ops/gemm_preshuffle/gemm_preshuffle_benchmark.py . --problem-sizes "1024,1024,1024" \
@@ -1664,15 +1664,15 @@ pipeline {
-D CMAKE_CXX_COMPILER="${build_compiler()}" \
-D CMAKE_BUILD_TYPE=Release \
-D GPU_TARGETS="gfx942" \
-D GEMM_DATATYPE="fp8;fp16" \
-D GEMM_LAYOUT="rcr;rrr;crr;ccr" \
-D GEMM_UNIVERSAL_DATATYPE="fp8;fp16" \
-D GEMM_UNIVERSAL_LAYOUT="rcr;rrr;crr;ccr" \
-D GEMM_MULTI_D_DATATYPE="fp16" \
-D GEMM_MULTI_D_LAYOUT="rcrr;rrrr;crrr;ccrr" \
-D GEMM_PRESHUFFLE_DATATYPE="fp16;fp8;bf16;bf8" \
-D GEMM_PRESHUFFLE_LAYOUT="rcr" \
-DCMAKE_CXX_FLAGS=" -O3 " .. && \
ninja -j64 benchmark_gemm_all && \
python3 ../tile_engine/ops/gemm/gemm_benchmark.py . --problem-sizes "1024,1024,1024" \
ninja -j64 benchmark_gemm_universal_all && \
python3 ../tile_engine/ops_new/gemm/gemm_universal/gemm_benchmark.py . --problem-sizes "1024,1024,1024" \
--warmup 5 --repeat 5 --verbose --json results.json && \
ninja -j64 benchmark_gemm_preshuffle_all && \
python3 ../tile_engine/ops/gemm_preshuffle/gemm_preshuffle_benchmark.py . --problem-sizes "1024,1024,1024" \

0
tile_engine/ops_new/gemm/gemm_multi_d/gemm_multi_d_instance_builder.py
View File

21
tile_engine/ops_new/gemm/gemm_preshuffle/gemm_preshuffle_instance_builder.py
View File

@@ -1,21 +0,0 @@
# class GemmPreshuffleKernelBuilder(BaseGemmKernelBuilder):
# def _get_kernel_prefix(self):
# return "gemm_preshuffle"
# def _get_pipeline_maps(self):
# # Return preshuffle-specific pipeline maps
# def _generate_kernel_instance(self, tile_config, trait_combo, k_block_per_cu, permute_n, is_header=True):
# # Preshuffle-specific kernel generation logic
# def _get_file_names(self):
# return "gemm_preshuffle_kernel_count.txt", "gemm_preshuffle_kernel_list.txt"
# def _get_cmake_function_name(self):
# return "create_individual_gemm_preshuffle_target"
# def _validate_layout_constraints(self, layout_parts):
# # Only allow rcr layout for preshuffle
# def _get_additional_validation_constraints(self, tile_config):
# # Preshuffle-specific validation (permute_n constraint)