From 4c0b5201eb0c002f588dc9fbbe761f61047147aa Mon Sep 17 00:00:00 2001 From: "assistant-librarian[bot]" Date: Tue, 14 Oct 2025 12:17:41 +0000 Subject: [PATCH] Merge commit '589e242eda730958b36c4f78bfad1991c499b0d2' into develop --- tile_engine/ops/gemm/gemm_instance_builder.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tile_engine/ops/gemm/gemm_instance_builder.py b/tile_engine/ops/gemm/gemm_instance_builder.py index c2214da613..0dc9fffedb 100644 --- a/tile_engine/ops/gemm/gemm_instance_builder.py +++ b/tile_engine/ops/gemm/gemm_instance_builder.py @@ -450,11 +450,11 @@ struct SelectedKernel {{ static constexpr ck_tile::index_t WarpTileK = {tile_config["warp_tile_k"]}; // Traits - static constexpr bool kPadM = {"true" if pad_m == "true" else "false"}; - static constexpr bool kPadN = {"true" if pad_n == "true" else "false"}; - static constexpr bool kPadK = {"true" if pad_k == "true" else "false"}; + static constexpr bool kPadM = {"true" if pad_m in [True, "true"] else "false"}; + static constexpr bool kPadN = {"true" if pad_n in [True, "true"] else "false"}; + static constexpr bool kPadK = {"true" if pad_k in [True, "true"] else "false"}; static constexpr bool TransposeC = false; - static constexpr bool UsePersistentKernel = {"true" if persistent == "true" else "false"}; + static constexpr bool UsePersistentKernel = {"true" if persistent in [True, "true"] else "false"}; static constexpr bool DoubleSmemBuffer = {"true" if pipeline == "compv4" else "false"}; static constexpr bool UseStructuredSparsity = false; static constexpr bool Preshuffle = false; @@ -576,7 +576,7 @@ struct SelectedKernel {{ }} // Get grid and block sizes - const dim3 grids = {"GemmKernel::MaxOccupancyGridSize(stream)" if persistent == "true" else "GemmKernel::GridSize(args.M, args.N, args.k_batch)"}; + const dim3 grids = {"GemmKernel::MaxOccupancyGridSize(stream)" if persistent in [True, "true"] else "GemmKernel::GridSize(args.M, args.N, args.k_batch)"}; const dim3 blocks = GemmKernel::BlockSize(); if(stream.log_level_ > 0) {{