From 20d5c905e26aa3eb90e37a52dbe347262e6ca655 Mon Sep 17 00:00:00 2001 From: Prabindh Sundareson Date: Wed, 21 Oct 2026 11:06:04 +0000 Subject: [PATCH] [fix] Serialisation of layout for inductor torch.compile emits incorrect syntax on current code LAYOUT= instead of LAYOUT=0 Tested on PyTorch 2.12.0+cu130 and Triton 3.7.0 Signed-off-by: Prabindh Sundareson --- bionemo_ir/dsl_kernels/triton/fused_layer_norm_transpose.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bionemo_ir/dsl_kernels/triton/fused_layer_norm_transpose.py b/bionemo_ir/dsl_kernels/triton/fused_layer_norm_transpose.py index 85648717..76077ee1 100644 --- a/bionemo_ir/dsl_kernels/triton/fused_layer_norm_transpose.py +++ b/bionemo_ir/dsl_kernels/triton/fused_layer_norm_transpose.py @@ -608,7 +608,7 @@ def _launch_layer_norm_transpose( HAS_WEIGHT=has_weight, HAS_BIAS=has_bias, RMS_NORM=rms_norm, - LAYOUT=layout, + LAYOUT=int(layout), SINGLE_TILE=single_tile, NEEDS_INT64=_needs_int64(B, N, D, tile_d), N_IN=0 if token_pad is None else token_pad.tokens_in,