From 1a07c853015d83e3662e85bdb9397b20ff5a8670 Mon Sep 17 00:00:00 2001 From: ThruptiRajLakshmanaGowda Date: Tue, 25 Nov 2025 23:46:00 +0000 Subject: [PATCH] Partial Progress : Jenkins change to run op_new --- Jenkinsfile | 16 +++++++------- .../gemm_multi_d_instance_builder.py | 0 .../gemm_preshuffle_instance_builder.py | 21 ------------------- 3 files changed, 8 insertions(+), 29 deletions(-) delete mode 100644 tile_engine/ops_new/gemm/gemm_multi_d/gemm_multi_d_instance_builder.py delete mode 100644 tile_engine/ops_new/gemm/gemm_preshuffle/gemm_preshuffle_instance_builder.py diff --git a/Jenkinsfile b/Jenkinsfile index 80db99684a..75e423734f 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -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" \ diff --git a/tile_engine/ops_new/gemm/gemm_multi_d/gemm_multi_d_instance_builder.py b/tile_engine/ops_new/gemm/gemm_multi_d/gemm_multi_d_instance_builder.py deleted file mode 100644 index e69de29bb2..0000000000 diff --git a/tile_engine/ops_new/gemm/gemm_preshuffle/gemm_preshuffle_instance_builder.py b/tile_engine/ops_new/gemm/gemm_preshuffle/gemm_preshuffle_instance_builder.py deleted file mode 100644 index 88d98026fd..0000000000 --- a/tile_engine/ops_new/gemm/gemm_preshuffle/gemm_preshuffle_instance_builder.py +++ /dev/null @@ -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)