Add files using upload-large-folder tool

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/__grp__triton_red_fused_argmax_1.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_red_fused_argmax_1.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.source", "triton_red_fused_argmax_1.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ttir", "triton_red_fused_argmax_1.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ttgir", "triton_red_fused_argmax_1.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.llir", "triton_red_fused_argmax_1.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ptx", "triton_red_fused_argmax_1.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.cubin", "triton_red_fused_argmax_1.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.json

@@ -0,0 +1 @@
+{"hash": "d764c4de3a434d91651ee340a32edffa72e41bcb0125e69cee815d3536f2f3ce", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 8, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 256, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_red_fused_argmax_1"}

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.llir

@@ -0,0 +1,206 @@
+; ModuleID = 'LLVMDialectModule'
+source_filename = "LLVMDialectModule"
+target datalayout = "e-p3:32:32-p4:32:32-p5:32:32-p6:32:32-p7:32:32-i64:64-i128:128-v16:16-v32:32-n16:32:64"
+
+@global_smem = external local_unnamed_addr addrspace(3) global [0 x i8], align 16
+
+; Function Attrs: nounwind
+define ptx_kernel void @triton_red_fused_argmax_1(ptr addrspace(1) %0, ptr addrspace(1) %1, i64 %2, i64 %3, i32 %4, i32 %5, ptr addrspace(1) readnone captures(none) %6, ptr addrspace(1) readnone captures(none) %7) local_unnamed_addr #0 !dbg !4 {
+  %9 = tail call i32 @llvm.nvvm.read.ptx.sreg.ctaid.x(), !dbg !7
+  %10 = shl i32 %9, 6, !dbg !8
+  %11 = tail call i32 @llvm.nvvm.read.ptx.sreg.tid.x(), !dbg !9
+  %12 = and i32 %11, 252, !dbg !9
+  %13 = lshr exact i32 %12, 2, !dbg !9
+  %14 = or disjoint i32 %13, %10, !dbg !10
+  %15 = icmp slt i32 %14, %4, !dbg !11
+  %16 = and i32 %11, 3, !dbg !12
+  %17 = sext i32 %14 to i64, !dbg !13
+  %.frozen = freeze i64 %2, !dbg !14
+  %18 = sdiv i64 %17, %.frozen, !dbg !14
+  %19 = mul i64 %18, %.frozen, !dbg !13
+  %.decomposed = sub i64 %17, %19, !dbg !13
+  %20 = mul i64 %18, %3, !dbg !15
+  %.idx = mul nsw i64 %.decomposed, 128000
+  %21 = getelementptr i8, ptr addrspace(1) %0, i64 %.idx
+  %invariant.gep = getelementptr float, ptr addrspace(1) %21, i64 %20, !dbg !16
+  %.fr = freeze i1 %15
+  %22 = zext nneg i32 %16 to i64, !dbg !16
+  br i1 %.fr, label %.split.us, label %.split.preheader
+
+.split.preheader:                                 ; preds = %8
+  %invariant.gep11 = getelementptr float, ptr addrspace(1) %invariant.gep, i64 %22, !dbg !16
+  br label %.split, !dbg !16
+
+.split.us:                                        ; preds = %8, %.split.us
+  %indvars.iv7 = phi i64 [ %indvars.iv.next8, %.split.us ], [ 0, %8 ]
+  %23 = phi i32 [ %44, %.split.us ], [ 2147483647, %8 ]
+  %24 = phi float [ %42, %.split.us ], [ 0xFFF0000000000000, %8 ]
+  %25 = or disjoint i64 %indvars.iv7, %22, !dbg !17
+  %gep.us = getelementptr float, ptr addrspace(1) %invariant.gep, i64 %25, !dbg !18
+  %26 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_first.b64 $0, 1.0;", "=l"() #4, !dbg !19
+  %27 = tail call i32 asm sideeffect "mov.u32 $0, $1;\0A\09@$4 ld.global.L1::evict_first.L2::cache_hint.b32 { $0 }, [ $2 + 0 ], $3;", "=r,r,l,l,b"(i32 0, ptr addrspace(1) %gep.us, i64 %26, i1 true) #4, !dbg !19
+  %28 = bitcast i32 %27 to float, !dbg !19
+  %29 = fcmp ogt float %24, %28, !dbg !20
+  %30 = fcmp oeq float %24, %28, !dbg !24
+  %31 = fcmp uno float %24, 0.000000e+00, !dbg !25
+  %32 = fcmp uno float %28, 0.000000e+00, !dbg !26
+  %33 = xor i1 %32, true, !dbg !27
+  %34 = and i1 %31, %33, !dbg !28
+  %35 = or i1 %29, %34, !dbg !29
+  %36 = and i1 %31, %32, !dbg !30
+  %37 = or i1 %30, %36, !dbg !31
+  %38 = sext i32 %23 to i64, !dbg !32
+  %39 = icmp sgt i64 %25, %38, !dbg !32
+  %40 = and i1 %39, %37, !dbg !33
+  %41 = or i1 %35, %40, !dbg !34
+  %42 = select i1 %41, float %24, float %28, !dbg !35
+  %43 = trunc nuw nsw i64 %25 to i32, !dbg !36
+  %44 = select i1 %41, i32 %23, i32 %43, !dbg !36
+  %indvars.iv.next8 = add nuw nsw i64 %indvars.iv7, 4, !dbg !16
+  %45 = icmp samesign ult i64 %indvars.iv7, 31996, !dbg !16
+  br i1 %45, label %.split.us, label %.split3.us, !dbg !16
+
+.split:                                           ; preds = %.split.preheader, %.split
+  %indvars.iv = phi i64 [ 0, %.split.preheader ], [ %indvars.iv.next, %.split ]
+  %gep12 = getelementptr float, ptr addrspace(1) %invariant.gep11, i64 %indvars.iv, !dbg !18
+  %46 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_first.b64 $0, 1.0;", "=l"() #4, !dbg !19
+  %47 = tail call i32 asm sideeffect "mov.u32 $0, $1;\0A\09@$4 ld.global.L1::evict_first.L2::cache_hint.b32 { $0 }, [ $2 + 0 ], $3;", "=r,r,l,l,b"(i32 0, ptr addrspace(1) %gep12, i64 %46, i1 false) #4, !dbg !19
+  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 4, !dbg !16
+  %48 = icmp samesign ult i64 %indvars.iv, 31996, !dbg !16
+  br i1 %48, label %.split, label %.split3.us, !dbg !16
+
+.split3.us:                                       ; preds = %.split, %.split.us
+  %.us-phi = phi float [ %42, %.split.us ], [ 0xFFF0000000000000, %.split ], !dbg !9
+  %.us-phi4 = phi i32 [ %44, %.split.us ], [ 2147483647, %.split ], !dbg !9
+  %49 = and i32 %11, 63, !dbg !9
+  %50 = or disjoint i32 %10, %49, !dbg !10
+  %51 = icmp slt i32 %50, %4, !dbg !11
+  %52 = bitcast float %.us-phi to i32, !dbg !37
+  %53 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %52, i32 2, i32 31), !dbg !37
+  %54 = bitcast i32 %53 to float, !dbg !37
+  %55 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %.us-phi4, i32 2, i32 31), !dbg !37
+  %56 = fcmp ogt float %.us-phi, %54, !dbg !39
+  %57 = fcmp oeq float %.us-phi, %54, !dbg !40
+  %58 = fcmp uno float %.us-phi, 0.000000e+00, !dbg !41
+  %59 = fcmp uno float %54, 0.000000e+00, !dbg !42
+  %60 = xor i1 %59, true, !dbg !43
+  %61 = and i1 %58, %60, !dbg !44
+  %62 = or i1 %56, %61, !dbg !45
+  %63 = and i1 %58, %59, !dbg !46
+  %64 = or i1 %57, %63, !dbg !47
+  %65 = icmp slt i32 %.us-phi4, %55, !dbg !48
+  %66 = and i1 %65, %64, !dbg !49
+  %67 = or i1 %62, %66, !dbg !50
+  %68 = select i1 %67, float %.us-phi, float %54, !dbg !51
+  %69 = select i1 %67, i32 %.us-phi4, i32 %55, !dbg !52
+  %70 = bitcast float %68 to i32, !dbg !37
+  %71 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %70, i32 1, i32 31), !dbg !37
+  %72 = bitcast i32 %71 to float, !dbg !37
+  %73 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %69, i32 1, i32 31), !dbg !37
+  %74 = fcmp ogt float %68, %72, !dbg !39
+  %75 = fcmp oeq float %68, %72, !dbg !40
+  %76 = fcmp uno float %68, 0.000000e+00, !dbg !41
+  %77 = fcmp uno float %72, 0.000000e+00, !dbg !42
+  %78 = xor i1 %77, true, !dbg !43
+  %79 = and i1 %76, %78, !dbg !44
+  %80 = or i1 %74, %79, !dbg !45
+  %81 = and i1 %77, %76, !dbg !46
+  %82 = or i1 %75, %81, !dbg !47
+  %83 = icmp slt i32 %69, %73, !dbg !48
+  %84 = and i1 %83, %82, !dbg !49
+  %85 = or i1 %80, %84, !dbg !50
+  %86 = select i1 %85, i32 %69, i32 %73, !dbg !52
+  %87 = sext i32 %50 to i64, !dbg !53
+  %88 = getelementptr i64, ptr addrspace(1) %1, i64 %87, !dbg !53
+  %89 = getelementptr inbounds nuw i8, ptr addrspace(3) @global_smem, i32 %12, !dbg !54
+  %90 = insertelement <1 x i32> poison, i32 %86, i64 0, !dbg !54
+  store <1 x i32> %90, ptr addrspace(3) %89, align 4, !dbg !54
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !54
+  %91 = shl nuw nsw i32 %49, 2, !dbg !54
+  %92 = getelementptr inbounds nuw i8, ptr addrspace(3) @global_smem, i32 %91, !dbg !54
+  %93 = load i32, ptr addrspace(3) %92, align 4, !dbg !54
+  %94 = sext i32 %93 to i64, !dbg !54
+  %95 = and i32 %11, 192, !dbg !54
+  %96 = icmp eq i32 %95, 0, !dbg !54
+  %97 = and i1 %96, %51, !dbg !54
+  tail call void asm sideeffect "@$2 st.global.b64 [ $1 + 0 ], { $0 };", "l,l,b"(i64 %94, ptr addrspace(1) %88, i1 %97) #4, !dbg !54
+  ret void, !dbg !55
+}
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 2147483647) i32 @llvm.nvvm.read.ptx.sreg.ctaid.x() #1
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 1024) i32 @llvm.nvvm.read.ptx.sreg.tid.x() #1
+
+; Function Attrs: convergent nocallback nounwind memory(inaccessiblemem: readwrite)
+declare i32 @llvm.nvvm.shfl.sync.bfly.i32(i32, i32, i32, i32) #2
+
+; Function Attrs: convergent nocallback nounwind
+declare void @llvm.nvvm.barrier.cta.sync.aligned.all(i32) #3
+
+attributes #0 = { nounwind "nvvm.reqntid"="256" }
+attributes #1 = { mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #2 = { convergent nocallback nounwind memory(inaccessiblemem: readwrite) }
+attributes #3 = { convergent nocallback nounwind }
+attributes #4 = { nounwind }
+
+!llvm.dbg.cu = !{!0}
+!llvm.module.flags = !{!2, !3}
+
+!0 = distinct !DICompileUnit(language: DW_LANG_C, file: !1, producer: "triton", isOptimized: true, runtimeVersion: 0, emissionKind: LineTablesOnly)
+!1 = !DIFile(filename: "ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py", directory: "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu")
+!2 = !{i32 2, !"Debug Info Version", i32 3}
+!3 = !{i32 4, !"nvvm-reflect-ftz", i32 1}
+!4 = distinct !DISubprogram(name: "triton_red_fused_argmax_1", linkageName: "triton_red_fused_argmax_1", scope: !1, file: !1, line: 18, type: !5, scopeLine: 18, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0)
+!5 = !DISubroutineType(cc: DW_CC_normal, types: !6)
+!6 = !{}
+!7 = !DILocation(line: 22, column: 28, scope: !4)
+!8 = !DILocation(line: 22, column: 33, scope: !4)
+!9 = !DILocation(line: 23, column: 44, scope: !4)
+!10 = !DILocation(line: 23, column: 23, scope: !4)
+!11 = !DILocation(line: 24, column: 21, scope: !4)
+!12 = !DILocation(line: 25, column: 37, scope: !4)
+!13 = !DILocation(line: 27, column: 19, scope: !4)
+!14 = !DILocation(line: 28, column: 19, scope: !4)
+!15 = !DILocation(line: 38, column: 56, scope: !4)
+!16 = !DILocation(line: 32, column: 40, scope: !4)
+!17 = !DILocation(line: 33, column: 31, scope: !4)
+!18 = !DILocation(line: 38, column: 34, scope: !4)
+!19 = !DILocation(line: 38, column: 61, scope: !4)
+!20 = !DILocation(line: 144, column: 21, scope: !21, inlinedAt: !23)
+!21 = distinct !DILexicalBlockFile(scope: !4, file: !22, discriminator: 0)
+!22 = !DIFile(filename: "triton_helpers.py", directory: "/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime")
+!23 = !DILocation(line: 41, column: 38, scope: !4)
+!24 = !DILocation(line: 145, column: 23, scope: !21, inlinedAt: !23)
+!25 = !DILocation(line: 147, column: 29, scope: !21, inlinedAt: !23)
+!26 = !DILocation(line: 148, column: 29, scope: !21, inlinedAt: !23)
+!27 = !DILocation(line: 149, column: 31, scope: !21, inlinedAt: !23)
+!28 = !DILocation(line: 149, column: 27, scope: !21, inlinedAt: !23)
+!29 = !DILocation(line: 149, column: 16, scope: !21, inlinedAt: !23)
+!30 = !DILocation(line: 151, column: 27, scope: !21, inlinedAt: !23)
+!31 = !DILocation(line: 151, column: 17, scope: !21, inlinedAt: !23)
+!32 = !DILocation(line: 154, column: 31, scope: !21, inlinedAt: !23)
+!33 = !DILocation(line: 154, column: 21, scope: !21, inlinedAt: !23)
+!34 = !DILocation(line: 154, column: 12, scope: !21, inlinedAt: !23)
+!35 = !DILocation(line: 155, column: 35, scope: !21, inlinedAt: !23)
+!36 = !DILocation(line: 155, column: 69, scope: !21, inlinedAt: !23)
+!37 = !DILocation(line: 165, column: 42, scope: !21, inlinedAt: !38)
+!38 = !DILocation(line: 45, column: 75, scope: !4)
+!39 = !DILocation(line: 144, column: 21, scope: !21, inlinedAt: !38)
+!40 = !DILocation(line: 145, column: 23, scope: !21, inlinedAt: !38)
+!41 = !DILocation(line: 147, column: 29, scope: !21, inlinedAt: !38)
+!42 = !DILocation(line: 148, column: 29, scope: !21, inlinedAt: !38)
+!43 = !DILocation(line: 149, column: 31, scope: !21, inlinedAt: !38)
+!44 = !DILocation(line: 149, column: 27, scope: !21, inlinedAt: !38)
+!45 = !DILocation(line: 149, column: 16, scope: !21, inlinedAt: !38)
+!46 = !DILocation(line: 151, column: 27, scope: !21, inlinedAt: !38)
+!47 = !DILocation(line: 151, column: 17, scope: !21, inlinedAt: !38)
+!48 = !DILocation(line: 154, column: 31, scope: !21, inlinedAt: !38)
+!49 = !DILocation(line: 154, column: 21, scope: !21, inlinedAt: !38)
+!50 = !DILocation(line: 154, column: 12, scope: !21, inlinedAt: !38)
+!51 = !DILocation(line: 155, column: 35, scope: !21, inlinedAt: !38)
+!52 = !DILocation(line: 155, column: 69, scope: !21, inlinedAt: !38)
+!53 = !DILocation(line: 47, column: 25, scope: !4)
+!54 = !DILocation(line: 47, column: 36, scope: !4)
+!55 = !DILocation(line: 47, column: 4, scope: !4)

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ptx

@@ -0,0 +1,490 @@
+//
+// Generated by LLVM NVPTX Back-End
+//
+
+.version 8.7
+.target sm_90a
+.address_size 64
+
+	// .globl	triton_red_fused_argmax_1 // -- Begin function triton_red_fused_argmax_1
+.extern .shared .align 16 .b8 global_smem[];
+                                        // @triton_red_fused_argmax_1
+.visible .entry triton_red_fused_argmax_1(
+	.param .u64 .ptr .global .align 1 triton_red_fused_argmax_1_param_0,
+	.param .u64 .ptr .global .align 1 triton_red_fused_argmax_1_param_1,
+	.param .u64 triton_red_fused_argmax_1_param_2,
+	.param .u64 triton_red_fused_argmax_1_param_3,
+	.param .u32 triton_red_fused_argmax_1_param_4,
+	.param .u32 triton_red_fused_argmax_1_param_5,
+	.param .u64 .ptr .global .align 1 triton_red_fused_argmax_1_param_6,
+	.param .u64 .ptr .global .align 1 triton_red_fused_argmax_1_param_7
+)
+.reqntid 256
+{
+	.reg .pred 	%p<39>;
+	.reg .b32 	%r<55>;
+	.reg .b64 	%rd<54>;
+	.loc	1 18 0                          // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:18:0
+$L__func_begin0:
+	.loc	1 18 0                          // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:18:0
+
+// %bb.0:
+	ld.param.b32 	%r12, [triton_red_fused_argmax_1_param_4];
+$L__tmp0:
+	.loc	1 22 28                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:22:28
+	mov.u32 	%r13, %ctaid.x;
+	.loc	1 22 33                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:22:33
+	shl.b32 	%r1, %r13, 6;
+	ld.param.b64 	%rd20, [triton_red_fused_argmax_1_param_2];
+	.loc	1 23 44                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:23:44
+	mov.u32 	%r2, %tid.x;
+	bfe.u32 	%r4, %r2, 2, 6;
+	.loc	1 23 23                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:23:23
+	or.b32 	%r14, %r4, %r1;
+	.loc	1 25 37                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:25:37
+	and.b32 	%r5, %r2, 3;
+	.loc	1 27 19                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:27:19
+	cvt.s64.s32 	%rd1, %r14;
+	.loc	1 28 19                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:28:19
+	or.b64 	%rd21, %rd1, %rd20;
+	and.b64 	%rd22, %rd21, -4294967296;
+	setp.ne.b64 	%p1, %rd22, 0;
+	cvt.u32.u64 	%r50, %rd1;
+	@%p1 bra 	$L__BB0_2;
+	bra.uni 	$L__BB0_1;
+$L__BB0_2:
+	div.s64 	%rd49, %rd1, %rd20;
+	bra.uni 	$L__BB0_3;
+$L__BB0_1:
+	cvt.u32.u64 	%r15, %rd20;
+	div.u32 	%r17, %r50, %r15;
+	cvt.u64.u32 	%rd49, %r17;
+$L__BB0_3:
+	.loc	1 0 19                          // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:0:19
+	ld.param.b64 	%rd19, [triton_red_fused_argmax_1_param_3];
+	ld.param.b64 	%rd18, [triton_red_fused_argmax_1_param_1];
+	ld.param.b64 	%rd17, [triton_red_fused_argmax_1_param_0];
+	and.b32 	%r3, %r2, 252;
+	.loc	1 32 40                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:32:40
+	cvt.u64.u32 	%rd6, %r5;
+	setp.ge.s32 	%p2, %r50, %r12;
+	@%p2 bra 	$L__BB0_6;
+// %bb.4:                               // %.split.us.preheader
+	shl.b64 	%rd35, %rd19, 2;
+	mul.lo.s64 	%rd36, %rd20, 128000;
+	sub.s64 	%rd37, %rd35, %rd36;
+	mul.lo.s64 	%rd38, %rd49, %rd37;
+	add.s32 	%r26, %r1, %r4;
+	mad.wide.s32 	%rd39, %r26, 128000, %rd38;
+	shl.b64 	%rd40, %rd6, 2;
+	add.s64 	%rd41, %rd39, %rd40;
+	add.s64 	%rd50, %rd17, %rd41;
+	mov.b32 	%r53, 0fFF800000;
+	mov.b32 	%r54, 2147483647;
+	mov.b64 	%rd51, 0;
+$L__BB0_5:                              // %.split.us
+                                        // =>This Inner Loop Header: Depth=1
+	.loc	1 38 34                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:38:34
+	add.s64 	%rd45, %rd6, %rd51;
+	.loc	1 38 61                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:38:61
+	// begin inline asm
+	mov.u64 %rd42, 0x0;
+	createpolicy.fractional.L2::evict_first.b64 %rd42, 1.0;
+	// end inline asm
+	mov.b32 	%r28, 0;
+	mov.pred 	%p5, -1;
+	// begin inline asm
+	mov.u32 %r27, %r28;
+	@%p5 ld.global.L1::evict_first.L2::cache_hint.b32 { %r27 }, [ %rd50 + 0 ], %rd42;
+	// end inline asm
+$L__tmp1:
+	.loc	2 144 21                        // triton_helpers.py:144:21 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	setp.gt.f32 	%p6, %r53, %r27;
+	.loc	2 145 23                        // triton_helpers.py:145:23 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	setp.eq.f32 	%p7, %r53, %r27;
+	.loc	2 147 29                        // triton_helpers.py:147:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	setp.nan.f32 	%p8, %r53, %r53;
+	.loc	2 148 29                        // triton_helpers.py:148:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	setp.nan.f32 	%p9, %r27, %r27;
+	setp.num.f32 	%p10, %r27, %r27;
+	.loc	2 149 27                        // triton_helpers.py:149:27 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	and.pred 	%p11, %p8, %p10;
+	.loc	2 149 16                        // triton_helpers.py:149:16 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	or.pred 	%p12, %p6, %p11;
+	.loc	2 151 27                        // triton_helpers.py:151:27 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	and.pred 	%p13, %p8, %p9;
+	.loc	2 151 17                        // triton_helpers.py:151:17 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	or.pred 	%p14, %p7, %p13;
+	.loc	2 154 31                        // triton_helpers.py:154:31 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	cvt.s64.s32 	%rd46, %r54;
+	setp.gt.s64 	%p15, %rd45, %rd46;
+	.loc	2 154 21                        // triton_helpers.py:154:21 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	and.pred 	%p16, %p15, %p14;
+	.loc	2 154 12                        // triton_helpers.py:154:12 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	or.pred 	%p17, %p12, %p16;
+	.loc	2 155 35                        // triton_helpers.py:155:35 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	selp.f32 	%r53, %r53, %r27, %p17;
+	cvt.u32.u64 	%r29, %rd45;
+	.loc	2 155 69                        // triton_helpers.py:155:69 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:41:38 ]
+	selp.b32 	%r54, %r54, %r29, %p17;
+$L__tmp2:
+	.loc	1 32 40                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:32:40
+	add.s64 	%rd11, %rd51, 4;
+	add.s64 	%rd50, %rd50, 16;
+	setp.lt.u64 	%p18, %rd51, 31996;
+	mov.b64 	%rd51, %rd11;
+	@%p18 bra 	$L__BB0_5;
+	bra.uni 	$L__BB0_8;
+$L__BB0_6:                              // %.split.preheader
+	shl.b64 	%rd24, %rd19, 2;
+	mul.lo.s64 	%rd25, %rd20, 128000;
+	sub.s64 	%rd26, %rd24, %rd25;
+	mul.lo.s64 	%rd27, %rd49, %rd26;
+	add.s32 	%r19, %r1, %r4;
+	mad.wide.s32 	%rd28, %r19, 128000, %rd27;
+	shl.b64 	%rd29, %rd6, 2;
+	add.s64 	%rd30, %rd28, %rd29;
+	add.s64 	%rd52, %rd17, %rd30;
+	mov.b64 	%rd53, -4;
+$L__BB0_7:                              // %.split
+                                        // =>This Inner Loop Header: Depth=1
+	.loc	1 38 61                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:38:61
+	// begin inline asm
+	mov.u64 %rd31, 0x0;
+	createpolicy.fractional.L2::evict_first.b64 %rd31, 1.0;
+	// end inline asm
+	mov.b32 	%r21, 0;
+	mov.pred 	%p3, 0;
+	// begin inline asm
+	mov.u32 %r20, %r21;
+	@%p3 ld.global.L1::evict_first.L2::cache_hint.b32 { %r20 }, [ %rd52 + 0 ], %rd31;
+	// end inline asm
+	.loc	1 32 40                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:32:40
+	add.s64 	%rd53, %rd53, 4;
+	add.s64 	%rd52, %rd52, 16;
+	setp.lt.u64 	%p4, %rd53, 31996;
+	mov.b32 	%r54, 2147483647;
+	mov.b32 	%r53, 0fFF800000;
+	@%p4 bra 	$L__BB0_7;
+$L__BB0_8:                              // %.split3.us
+	.loc	1 23 44                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:23:44
+	and.b32 	%r30, %r2, 63;
+	.loc	1 23 23                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:23:23
+	or.b32 	%r31, %r1, %r30;
+	.loc	1 24 21                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:24:21
+	setp.lt.s32 	%p20, %r31, %r12;
+$L__tmp3:
+	.loc	2 165 42                        // triton_helpers.py:165:42 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	shfl.sync.bfly.b32 	%r32, %r53, 2, 31, -1;
+	shfl.sync.bfly.b32 	%r33, %r54, 2, 31, -1;
+	.loc	2 144 21                        // triton_helpers.py:144:21 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.gt.f32 	%p21, %r53, %r32;
+	.loc	2 145 23                        // triton_helpers.py:145:23 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.eq.f32 	%p22, %r53, %r32;
+	.loc	2 147 29                        // triton_helpers.py:147:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.nan.f32 	%p23, %r53, %r53;
+	.loc	2 148 29                        // triton_helpers.py:148:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.nan.f32 	%p24, %r32, %r32;
+	setp.num.f32 	%p25, %r32, %r32;
+	.loc	2 149 27                        // triton_helpers.py:149:27 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	and.pred 	%p26, %p23, %p25;
+	.loc	2 149 16                        // triton_helpers.py:149:16 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	or.pred 	%p27, %p21, %p26;
+	.loc	2 151 27                        // triton_helpers.py:151:27 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	and.pred 	%p28, %p23, %p24;
+	.loc	2 151 17                        // triton_helpers.py:151:17 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	or.pred 	%p29, %p22, %p28;
+	.loc	2 154 31                        // triton_helpers.py:154:31 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.lt.s32 	%p30, %r54, %r33;
+	.loc	2 154 21                        // triton_helpers.py:154:21 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	and.pred 	%p31, %p30, %p29;
+	.loc	2 154 12                        // triton_helpers.py:154:12 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	or.pred 	%p32, %p27, %p31;
+	.loc	2 155 35                        // triton_helpers.py:155:35 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	selp.f32 	%r34, %r53, %r32, %p32;
+	.loc	2 155 69                        // triton_helpers.py:155:69 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	selp.b32 	%r35, %r54, %r33, %p32;
+	.loc	2 165 42                        // triton_helpers.py:165:42 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	shfl.sync.bfly.b32 	%r36, %r34, 1, 31, -1;
+	shfl.sync.bfly.b32 	%r37, %r35, 1, 31, -1;
+	.loc	2 144 21                        // triton_helpers.py:144:21 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.gt.f32 	%p33, %r34, %r36;
+	.loc	2 145 23                        // triton_helpers.py:145:23 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.eq.f32 	%p34, %r34, %r36;
+	.loc	2 147 29                        // triton_helpers.py:147:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.nan.f32 	%p35, %r34, %r34;
+	.loc	2 148 29                        // triton_helpers.py:148:29 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.nan.f32 	%p36, %r36, %r36;
+	.loc	2 154 31                        // triton_helpers.py:154:31 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	setp.lt.s32 	%p37, %r35, %r37;
+	.loc	2 155 69                        // triton_helpers.py:155:69 @[ ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:45:75 ]
+	selp.b32 	%r38, %r35, %r37, %p35;
+	selp.b32 	%r39, %r38, %r37, %p36;
+	selp.b32 	%r40, %r35, %r39, %p34;
+	selp.b32 	%r41, %r40, %r37, %p37;
+	selp.b32 	%r42, %r41, %r35, %p36;
+	selp.b32 	%r43, %r42, %r41, %p35;
+	selp.b32 	%r44, %r35, %r43, %p33;
+$L__tmp4:
+	.loc	1 47 25                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:47:25
+	mad.wide.s32 	%rd48, %r31, 8, %rd18;
+	.loc	1 47 36                         // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:47:36
+	mov.b32 	%r45, global_smem;
+	add.s32 	%r46, %r45, %r3;
+	st.shared.b32 	[%r46], %r44;
+	bar.sync 	0;
+	shl.b32 	%r47, %r30, 2;
+	add.s32 	%r48, %r45, %r47;
+	ld.shared.s32 	%rd47, [%r48];
+	and.b32 	%r49, %r2, 192;
+	setp.eq.b32 	%p38, %r49, 0;
+	and.pred 	%p19, %p38, %p20;
+	// begin inline asm
+	@%p19 st.global.b64 [ %rd48 + 0 ], { %rd47 };
+	// end inline asm
+	.loc	1 47 4                          // ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py:47:4
+	ret;
+$L__tmp5:
+$L__func_end0:
+                                        // -- End function
+}
+	.file	1 "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py"
+	.file	2 "/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py"
+	.section	.debug_abbrev
+	{
+.b8 1                                   // Abbreviation Code
+.b8 17                                  // DW_TAG_compile_unit
+.b8 1                                   // DW_CHILDREN_yes
+.b8 37                                  // DW_AT_producer
+.b8 8                                   // DW_FORM_string
+.b8 19                                  // DW_AT_language
+.b8 5                                   // DW_FORM_data2
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 16                                  // DW_AT_stmt_list
+.b8 6                                   // DW_FORM_data4
+.b8 27                                  // DW_AT_comp_dir
+.b8 8                                   // DW_FORM_string
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 2                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 0                                   // DW_CHILDREN_no
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 32                                  // DW_AT_inline
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 3                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 1                                   // DW_CHILDREN_yes
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 4                                   // Abbreviation Code
+.b8 29                                  // DW_TAG_inlined_subroutine
+.b8 0                                   // DW_CHILDREN_no
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 88                                  // DW_AT_call_file
+.b8 11                                  // DW_FORM_data1
+.b8 89                                  // DW_AT_call_line
+.b8 11                                  // DW_FORM_data1
+.b8 87                                  // DW_AT_call_column
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 0                                   // EOM(3)
+	}
+	.section	.debug_info
+	{
+.b32 234                                // Length of Unit
+.b8 2                                   // DWARF version number
+.b8 0
+.b32 .debug_abbrev                      // Offset Into Abbrev. Section
+.b8 8                                   // Address Size (in bytes)
+.b8 1                                   // Abbrev [1] 0xb:0xe3 DW_TAG_compile_unit
+.b8 116                                 // DW_AT_producer
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 0
+.b8 2                                   // DW_AT_language
+.b8 0
+.b8 99                                  // DW_AT_name
+.b8 101
+.b8 117
+.b8 105
+.b8 54
+.b8 113
+.b8 114
+.b8 98
+.b8 50
+.b8 116
+.b8 51
+.b8 108
+.b8 109
+.b8 122
+.b8 115
+.b8 51
+.b8 108
+.b8 106
+.b8 114
+.b8 113
+.b8 116
+.b8 99
+.b8 111
+.b8 109
+.b8 116
+.b8 52
+.b8 98
+.b8 50
+.b8 113
+.b8 54
+.b8 115
+.b8 118
+.b8 122
+.b8 111
+.b8 50
+.b8 52
+.b8 106
+.b8 54
+.b8 109
+.b8 109
+.b8 114
+.b8 121
+.b8 97
+.b8 105
+.b8 111
+.b8 118
+.b8 114
+.b8 54
+.b8 107
+.b8 112
+.b8 55
+.b8 121
+.b8 46
+.b8 112
+.b8 121
+.b8 0
+.b32 .debug_line                        // DW_AT_stmt_list
+.b8 47                                  // DW_AT_comp_dir
+.b8 119
+.b8 111
+.b8 114
+.b8 107
+.b8 115
+.b8 112
+.b8 97
+.b8 99
+.b8 101
+.b8 47
+.b8 104
+.b8 97
+.b8 110
+.b8 114
+.b8 117
+.b8 105
+.b8 47
+.b8 83
+.b8 112
+.b8 101
+.b8 99
+.b8 70
+.b8 111
+.b8 114
+.b8 103
+.b8 101
+.b8 45
+.b8 101
+.b8 120
+.b8 116
+.b8 47
+.b8 99
+.b8 97
+.b8 99
+.b8 104
+.b8 101
+.b8 47
+.b8 99
+.b8 111
+.b8 109
+.b8 112
+.b8 105
+.b8 108
+.b8 101
+.b8 100
+.b8 95
+.b8 107
+.b8 101
+.b8 114
+.b8 110
+.b8 101
+.b8 108
+.b8 115
+.b8 47
+.b8 101
+.b8 117
+.b8 0
+.b8 2                                   // Abbrev [2] 0x8b:0x1c DW_TAG_subprogram
+.b8 116                                 // DW_AT_name
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 95
+.b8 114
+.b8 101
+.b8 100
+.b8 95
+.b8 102
+.b8 117
+.b8 115
+.b8 101
+.b8 100
+.b8 95
+.b8 97
+.b8 114
+.b8 103
+.b8 109
+.b8 97
+.b8 120
+.b8 95
+.b8 49
+.b8 0
+.b8 1                                   // DW_AT_inline
+.b8 3                                   // Abbrev [3] 0xa7:0x46 DW_TAG_subprogram
+.b64 $L__func_begin0                    // DW_AT_low_pc
+.b64 $L__func_end0                      // DW_AT_high_pc
+.b32 139                                // DW_AT_abstract_origin
+.b8 4                                   // Abbrev [4] 0xbc:0x18 DW_TAG_inlined_subroutine
+.b32 139                                // DW_AT_abstract_origin
+.b64 $L__tmp1                           // DW_AT_low_pc
+.b64 $L__tmp2                           // DW_AT_high_pc
+.b8 1                                   // DW_AT_call_file
+.b8 41                                  // DW_AT_call_line
+.b8 38                                  // DW_AT_call_column
+.b8 4                                   // Abbrev [4] 0xd4:0x18 DW_TAG_inlined_subroutine
+.b32 139                                // DW_AT_abstract_origin
+.b64 $L__tmp3                           // DW_AT_low_pc
+.b64 $L__tmp4                           // DW_AT_high_pc
+.b8 1                                   // DW_AT_call_file
+.b8 45                                  // DW_AT_call_line
+.b8 75                                  // DW_AT_call_column
+.b8 0                                   // End Of Children Mark
+.b8 0                                   // End Of Children Mark
+	}
+	.section	.debug_macinfo	{	}

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.source

@@ -0,0 +1,323 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":18:0)
+#loc35 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":143:0)
+#loc47 = loc(unknown)
+#loc55 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":86:0)
+#loc59 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":63:0)
+#loc68 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":164:0)
+#loc72 = loc("in_ptr0"(#loc))
+#loc73 = loc("out_ptr0"(#loc))
+#loc74 = loc("ks0"(#loc))
+#loc75 = loc("ks1"(#loc))
+#loc76 = loc("xnumel"(#loc))
+#loc77 = loc("r0_numel"(#loc))
+#loc106 = loc("a_value"(#loc35))
+#loc107 = loc("a_index"(#loc35))
+#loc108 = loc("b_value"(#loc35))
+#loc109 = loc("b_index"(#loc35))
+#loc122 = loc("x"(#loc55))
+#loc123 = loc("x"(#loc59))
+#loc124 = loc("value"(#loc68))
+#loc125 = loc("index"(#loc68))
+module {
+  tt.func public @triton_red_fused_argmax_1(%in_ptr0: !tt.ptr<f32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr0: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 {tt.divisibility = 16 : i32} loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %r0_numel_0 = arith.constant 32000 : i32 loc(#loc78)
+    %xoffset = tt.get_program_id x : i32 loc(#loc79)
+    %xoffset_1 = arith.constant 64 : i32 loc(#loc80)
+    %xoffset_2 = arith.constant 64 : i32 loc(#loc80)
+    %xoffset_3 = arith.muli %xoffset, %xoffset_2 : i32 loc(#loc80)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32> loc(#loc81)
+    %xindex_4 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc82)
+    %xindex_5 = tt.splat %xoffset_3 : i32 -> tensor<64x1xi32> loc(#loc83)
+    %xindex_6 = arith.addi %xindex_5, %xindex_4 : tensor<64x1xi32> loc(#loc83)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32> loc(#loc84)
+    %xmask_7 = arith.cmpi slt, %xindex_6, %xmask : tensor<64x1xi32> loc(#loc84)
+    %r0_base = tt.make_range {end = 4 : i32, start = 0 : i32} : tensor<4xi32> loc(#loc85)
+    %r0_base_8 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<4xi32> -> tensor<1x4xi32> loc(#loc86)
+    %x0 = arith.extsi %xindex_6 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc87)
+    %x0_9 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc87)
+    %x0_10 = arith.remsi %x0, %x0_9 : tensor<64x1xi64> loc(#loc87)
+    %x1 = arith.extsi %xindex_6 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc88)
+    %x1_11 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc88)
+    %x1_12 = arith.divsi %x1, %x1_11 : tensor<64x1xi64> loc(#loc88)
+    %_tmp2 = arith.constant 0xFF800000 : f32 loc(#loc89)
+    %_tmp2_13 = arith.constant dense<0xFF800000> : tensor<64x4xf32> loc(#loc89)
+    %_tmp2_index = arith.constant 2147483647 : i32 loc(#loc90)
+    %_tmp2_index_14 = arith.constant dense<2147483647> : tensor<64x4xi32> loc(#loc90)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc14)
+    %c4_i32 = arith.constant 4 : i32 loc(#loc14)
+    %0 = arith.bitcast %c0_i32 : i32 to i32 loc(#loc14)
+    %1 = arith.bitcast %r0_numel_0 : i32 to i32 loc(#loc14)
+    %2 = arith.bitcast %c4_i32 : i32 to i32 loc(#loc14)
+    %3 = ub.poison : i32 loc(#loc14)
+    %_tmp2_index_15:2 = scf.for %r0_offset = %0 to %1 step %2 iter_args(%_tmp2_16 = %_tmp2_13, %_tmp2_index_17 = %_tmp2_index_14) -> (tensor<64x4xf32>, tensor<64x4xi32>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x4xi32> loc(#loc92)
+      %r0_index_18 = arith.addi %r0_index, %r0_base_8 : tensor<1x4xi32> loc(#loc92)
+      %r0_mask = arith.constant dense<32000> : tensor<1x4xi32> loc(#loc93)
+      %r0_mask_19 = arith.cmpi slt, %r0_index_18, %r0_mask : tensor<1x4xi32> loc(#loc93)
+      %tmp0 = arith.constant 32000 : i32 loc(#loc94)
+      %tmp0_20 = arith.constant 32000 : i64 loc(#loc94)
+      %tmp0_21 = arith.constant dense<32000> : tensor<64x1xi64> loc(#loc94)
+      %tmp0_22 = arith.muli %tmp0_21, %x0_10 : tensor<64x1xi64> loc(#loc94)
+      %tmp0_23 = arith.extsi %r0_index_18 : tensor<1x4xi32> to tensor<1x4xi64> loc(#loc95)
+      %tmp0_24 = tt.broadcast %tmp0_23 : tensor<1x4xi64> -> tensor<64x4xi64> loc(#loc95)
+      %tmp0_25 = tt.broadcast %tmp0_22 : tensor<64x1xi64> -> tensor<64x4xi64> loc(#loc95)
+      %tmp0_26 = arith.addi %tmp0_24, %tmp0_25 : tensor<64x4xi64> loc(#loc95)
+      %tmp0_27 = tt.splat %ks1 : i64 -> tensor<64x1xi64> loc(#loc96)
+      %tmp0_28 = arith.muli %tmp0_27, %x1_12 : tensor<64x1xi64> loc(#loc96)
+      %tmp0_29 = tt.broadcast %tmp0_28 : tensor<64x1xi64> -> tensor<64x4xi64> loc(#loc97)
+      %tmp0_30 = arith.addi %tmp0_26, %tmp0_29 : tensor<64x4xi64> loc(#loc97)
+      %tmp0_31 = tt.splat %in_ptr0 : !tt.ptr<f32> -> tensor<64x4x!tt.ptr<f32>> loc(#loc98)
+      %tmp0_32 = tt.addptr %tmp0_31, %tmp0_30 : tensor<64x4x!tt.ptr<f32>>, tensor<64x4xi64> loc(#loc98)
+      %tmp0_33 = tt.broadcast %r0_mask_19 : tensor<1x4xi1> -> tensor<64x4xi1> loc(#loc99)
+      %tmp0_34 = tt.broadcast %xmask_7 : tensor<64x1xi1> -> tensor<64x4xi1> loc(#loc99)
+      %tmp0_35 = arith.andi %tmp0_33, %tmp0_34 : tensor<64x4xi1> loc(#loc99)
+      %tmp0_36 = arith.constant 0.000000e+00 : f32 loc(#loc100)
+      %tmp0_37 = arith.constant dense<0.000000e+00> : tensor<64x4xf32> loc(#loc100)
+      %tmp0_38 = tt.load %tmp0_32, %tmp0_35, %tmp0_37 evictionPolicy = evict_first : tensor<64x4x!tt.ptr<f32>> loc(#loc100)
+      %8:2 = tt.call @torch._inductor.runtime.triton_helpers.maximum_with_index__fp32S64_4S_i32S64_4S_fp32S64_4S_i32S1_4S__(%_tmp2_16, %_tmp2_index_17, %tmp0_38, %r0_index_18) : (tensor<64x4xf32>, tensor<64x4xi32>, tensor<64x4xf32>, tensor<1x4xi32>) -> (tensor<64x4xf32>, tensor<64x4xi32>) loc(#loc24)
+      %_tmp2_39 = tt.broadcast %r0_mask_19 : tensor<1x4xi1> -> tensor<64x4xi1> loc(#loc101)
+      %_tmp2_40 = tt.broadcast %xmask_7 : tensor<64x1xi1> -> tensor<64x4xi1> loc(#loc101)
+      %_tmp2_41 = arith.andi %_tmp2_39, %_tmp2_40 : tensor<64x4xi1> loc(#loc101)
+      %_tmp2_42 = arith.select %_tmp2_41, %8#0, %_tmp2_16 : tensor<64x4xi1>, tensor<64x4xf32> loc(#loc102)
+      %_tmp2_index_43 = tt.broadcast %r0_mask_19 : tensor<1x4xi1> -> tensor<64x4xi1> loc(#loc103)
+      %_tmp2_index_44 = tt.broadcast %xmask_7 : tensor<64x1xi1> -> tensor<64x4xi1> loc(#loc103)
+      %_tmp2_index_45 = arith.andi %_tmp2_index_43, %_tmp2_index_44 : tensor<64x4xi1> loc(#loc103)
+      %_tmp2_index_46 = arith.select %_tmp2_index_45, %8#1, %_tmp2_index_17 : tensor<64x4xi1>, tensor<64x4xi32> loc(#loc104)
+      scf.yield %_tmp2_42, %_tmp2_index_46 : tensor<64x4xf32>, tensor<64x4xi32> loc(#loc29)
+    } loc(#loc126)
+    %4:2 = tt.call @"torch._inductor.runtime.triton_helpers.max_with_index__fp32S64_4S_i32S64_4S__(2,)cconstexpr_1_"(%_tmp2_index_15#0, %_tmp2_index_15#1) : (tensor<64x4xf32>, tensor<64x4xi32>) -> (tensor<64xf32>, tensor<64xi32>) loc(#loc30)
+    %tmp2 = tt.expand_dims %4#1 {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc105)
+    %5 = tt.splat %out_ptr0 : !tt.ptr<i64> -> tensor<64x1x!tt.ptr<i64>> loc(#loc32)
+    %6 = tt.addptr %5, %xindex_6 : tensor<64x1x!tt.ptr<i64>>, tensor<64x1xi32> loc(#loc32)
+    %7 = arith.extsi %tmp2 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc33)
+    tt.store %6, %7, %xmask_7 : tensor<64x1x!tt.ptr<i64>> loc(#loc33)
+    tt.return loc(#loc34)
+  } loc(#loc)
+  tt.func private @torch._inductor.runtime.triton_helpers.maximum_with_index__fp32S64_4S_i32S64_4S_fp32S64_4S_i32S1_4S__(%a_value: tensor<64x4xf32> loc("a_value"(#loc35)), %a_index: tensor<64x4xi32> loc("a_index"(#loc35)), %b_value: tensor<64x4xf32> loc("b_value"(#loc35)), %b_index: tensor<1x4xi32> loc("b_index"(#loc35))) -> (tensor<64x4xf32>, tensor<64x4xi32>) attributes {noinline = false} {
+    %mask = arith.cmpf ogt, %a_value, %b_value : tensor<64x4xf32> loc(#loc127)
+    %equal = arith.cmpf oeq, %a_value, %b_value : tensor<64x4xf32> loc(#loc128)
+    %0 = tt.call @torch._inductor.runtime.triton_helpers.is_floating__fp32S64_4S__(%a_value) : (tensor<64x4xf32>) -> i1 loc(#loc38)
+    %1:2 = scf.if %0 -> (tensor<64x4xi1>, tensor<64x4xi1>) {
+      %a_isnan = arith.cmpf une, %a_value, %a_value : tensor<64x4xf32> loc(#loc112)
+      %b_isnan = arith.cmpf une, %b_value, %b_value : tensor<64x4xf32> loc(#loc113)
+      %mask_4 = arith.constant true loc(#loc114)
+      %mask_5 = arith.constant dense<true> : tensor<64x4xi1> loc(#loc114)
+      %mask_6 = arith.xori %b_isnan, %mask_5 : tensor<64x4xi1> loc(#loc114)
+      %mask_7 = arith.andi %a_isnan, %mask_6 : tensor<64x4xi1> loc(#loc115)
+      %mask_8 = arith.ori %mask, %mask_7 : tensor<64x4xi1> loc(#loc129)
+      %equal_9 = arith.andi %a_isnan, %b_isnan : tensor<64x4xi1> loc(#loc117)
+      %equal_10 = arith.ori %equal, %equal_9 : tensor<64x4xi1> loc(#loc130)
+      scf.yield %mask_8, %equal_10 : tensor<64x4xi1>, tensor<64x4xi1> loc(#loc130)
+    } else {
+      scf.yield %mask, %equal : tensor<64x4xi1>, tensor<64x4xi1> loc(#loc47)
+    } loc(#loc39)
+    %mask_0 = tt.broadcast %b_index : tensor<1x4xi32> -> tensor<64x4xi32> loc(#loc119)
+    %mask_1 = arith.cmpi slt, %a_index, %mask_0 : tensor<64x4xi32> loc(#loc119)
+    %mask_2 = arith.andi %1#1, %mask_1 : tensor<64x4xi1> loc(#loc120)
+    %mask_3 = arith.ori %1#0, %mask_2 : tensor<64x4xi1> loc(#loc121)
+    %2 = arith.select %mask_3, %a_value, %b_value : tensor<64x4xi1>, tensor<64x4xf32> loc(#loc51)
+    %3 = tt.broadcast %b_index : tensor<1x4xi32> -> tensor<64x4xi32> loc(#loc52)
+    %4 = arith.select %mask_3, %a_index, %3 : tensor<64x4xi1>, tensor<64x4xi32> loc(#loc52)
+    tt.return %2, %4 : tensor<64x4xf32>, tensor<64x4xi32> loc(#loc53)
+  ^bb1:  // no predecessors
+    %5 = ub.poison : tensor<64x4xf32> loc(#loc54)
+    %6 = ub.poison : tensor<64x4xi32> loc(#loc54)
+    tt.return %5, %6 : tensor<64x4xf32>, tensor<64x4xi32> loc(#loc54)
+  } loc(#loc35)
+  tt.func private @torch._inductor.runtime.triton_helpers.is_floating__fp32S64_4S__(%x: tensor<64x4xf32> loc("x"(#loc55))) -> i1 attributes {noinline = false} {
+    %0 = tt.call @torch._inductor.runtime.triton_helpers.promote_to_tensor__fp32S64_4S__(%x) : (tensor<64x4xf32>) -> tensor<64x4xf32> loc(#loc56)
+    %true = arith.constant true loc(#loc57)
+    tt.return %true : i1 loc(#loc57)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : i1 loc(#loc58)
+    tt.return %1 : i1 loc(#loc58)
+  } loc(#loc55)
+  tt.func private @torch._inductor.runtime.triton_helpers.promote_to_tensor__fp32S64_4S__(%x: tensor<64x4xf32> loc("x"(#loc59))) -> tensor<64x4xf32> attributes {noinline = false} {
+    %0 = tt.call @"triton.language.standard.zeros____(0, 0)cconstexpr_1__(1,)cconstexpr_int1_"() : () -> tensor<1xi1> loc(#loc60)
+    %1 = arith.uitofp %0 : tensor<1xi1> to tensor<1xf32> loc(#loc61)
+    %2 = tt.expand_dims %1 {axis = 0 : i32} : tensor<1xf32> -> tensor<1x1xf32> loc(#loc61)
+    %3 = tt.broadcast %2 : tensor<1x1xf32> -> tensor<64x4xf32> loc(#loc61)
+    %4 = arith.addf %x, %3 : tensor<64x4xf32> loc(#loc61)
+    tt.return %4 : tensor<64x4xf32> loc(#loc62)
+  ^bb1:  // no predecessors
+    %5 = ub.poison : tensor<64x4xf32> loc(#loc63)
+    tt.return %5 : tensor<64x4xf32> loc(#loc63)
+  } loc(#loc59)
+  tt.func private @"triton.language.standard.zeros____(0, 0)cconstexpr_1__(1,)cconstexpr_int1_"() -> tensor<1xi1> attributes {noinline = false} {
+    %false = arith.constant false loc(#loc65)
+    %cst = arith.constant dense<false> : tensor<1xi1> loc(#loc65)
+    tt.return %cst : tensor<1xi1> loc(#loc66)
+  ^bb1:  // no predecessors
+    %0 = ub.poison : tensor<1xi1> loc(#loc67)
+    tt.return %0 : tensor<1xi1> loc(#loc67)
+  } loc(#loc64)
+  tt.func private @"torch._inductor.runtime.triton_helpers.max_with_index__fp32S64_4S_i32S64_4S__(2,)cconstexpr_1_"(%value: tensor<64x4xf32> loc("value"(#loc68)), %index: tensor<64x4xi32> loc("index"(#loc68))) -> (tensor<64xf32>, tensor<64xi32>) attributes {noinline = false} {
+    %0:2 = "tt.reduce"(%value, %index) <{axis = 1 : i32}> ({
+    ^bb0(%arg2: f32 loc(unknown), %arg3: i32 loc(unknown), %arg4: f32 loc(unknown), %arg5: i32 loc(unknown)):
+      %3:2 = tt.call @torch._inductor.runtime.triton_helpers.maximum_with_index__fp32_i32_fp32_i32__(%arg2, %arg3, %arg4, %arg5) : (f32, i32, f32, i32) -> (f32, i32) loc(#loc69)
+      tt.reduce.return %3#0, %3#1 : f32, i32 loc(#loc69)
+    }) : (tensor<64x4xf32>, tensor<64x4xi32>) -> (tensor<64xf32>, tensor<64xi32>) loc(#loc69)
+    tt.return %0#0, %0#1 : tensor<64xf32>, tensor<64xi32> loc(#loc70)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : tensor<64xf32> loc(#loc71)
+    %2 = ub.poison : tensor<64xi32> loc(#loc71)
+    tt.return %1, %2 : tensor<64xf32>, tensor<64xi32> loc(#loc71)
+  } loc(#loc68)
+  tt.func private @torch._inductor.runtime.triton_helpers.maximum_with_index__fp32_i32_fp32_i32__(%a_value: f32 loc("a_value"(#loc35)), %a_index: i32 loc("a_index"(#loc35)), %b_value: f32 loc("b_value"(#loc35)), %b_index: i32 loc("b_index"(#loc35))) -> (f32, i32) attributes {noinline = false} {
+    %mask = arith.cmpf ogt, %a_value, %b_value : f32 loc(#loc127)
+    %equal = arith.cmpf oeq, %a_value, %b_value : f32 loc(#loc128)
+    %0 = tt.call @torch._inductor.runtime.triton_helpers.is_floating__fp32__(%a_value) : (f32) -> i1 loc(#loc38)
+    %1:2 = scf.if %0 -> (i1, i1) {
+      %a_isnan = arith.cmpf une, %a_value, %a_value : f32 loc(#loc112)
+      %b_isnan = arith.cmpf une, %b_value, %b_value : f32 loc(#loc113)
+      %mask_3 = arith.constant true loc(#loc114)
+      %mask_4 = arith.xori %b_isnan, %mask_3 : i1 loc(#loc114)
+      %mask_5 = arith.andi %a_isnan, %mask_4 : i1 loc(#loc115)
+      %mask_6 = arith.ori %mask, %mask_5 : i1 loc(#loc129)
+      %equal_7 = arith.andi %a_isnan, %b_isnan : i1 loc(#loc117)
+      %equal_8 = arith.ori %equal, %equal_7 : i1 loc(#loc130)
+      scf.yield %mask_6, %equal_8 : i1, i1 loc(#loc130)
+    } else {
+      scf.yield %mask, %equal : i1, i1 loc(#loc47)
+    } loc(#loc39)
+    %mask_0 = arith.cmpi slt, %a_index, %b_index : i32 loc(#loc119)
+    %mask_1 = arith.andi %1#1, %mask_0 : i1 loc(#loc120)
+    %mask_2 = arith.ori %1#0, %mask_1 : i1 loc(#loc121)
+    %2 = arith.select %mask_2, %a_value, %b_value : f32 loc(#loc51)
+    %3 = arith.select %mask_2, %a_index, %b_index : i32 loc(#loc52)
+    tt.return %2, %3 : f32, i32 loc(#loc53)
+  ^bb1:  // no predecessors
+    %4 = ub.poison : f32 loc(#loc54)
+    %5 = ub.poison : i32 loc(#loc54)
+    tt.return %4, %5 : f32, i32 loc(#loc54)
+  } loc(#loc35)
+  tt.func private @torch._inductor.runtime.triton_helpers.is_floating__fp32__(%x: f32 loc("x"(#loc55))) -> i1 attributes {noinline = false} {
+    %0 = tt.call @torch._inductor.runtime.triton_helpers.promote_to_tensor__fp32__(%x) : (f32) -> tensor<1xf32> loc(#loc56)
+    %true = arith.constant true loc(#loc57)
+    tt.return %true : i1 loc(#loc57)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : i1 loc(#loc58)
+    tt.return %1 : i1 loc(#loc58)
+  } loc(#loc55)
+  tt.func private @torch._inductor.runtime.triton_helpers.promote_to_tensor__fp32__(%x: f32 loc("x"(#loc59))) -> tensor<1xf32> attributes {noinline = false} {
+    %0 = tt.call @"triton.language.standard.zeros____(0, 0)cconstexpr_1__(1,)cconstexpr_int1_"() : () -> tensor<1xi1> loc(#loc60)
+    %1 = arith.uitofp %0 : tensor<1xi1> to tensor<1xf32> loc(#loc61)
+    %2 = tt.splat %x : f32 -> tensor<1xf32> loc(#loc61)
+    %3 = arith.addf %2, %1 : tensor<1xf32> loc(#loc61)
+    tt.return %3 : tensor<1xf32> loc(#loc62)
+  ^bb1:  // no predecessors
+    %4 = ub.poison : tensor<1xf32> loc(#loc63)
+    tt.return %4 : tensor<1xf32> loc(#loc63)
+  } loc(#loc59)
+} loc(#loc)
+#loc1 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":19:15)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:36)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:44)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:23)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":24:21)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":25:27)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":25:37)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":27:19)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":28:19)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":29:55)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":30:58)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":32:40)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":33:31)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":34:29)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:47)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:41)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:56)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:52)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:34)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:71)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:61)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":41:38)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":43:35)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":43:54)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:41)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:66)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:8)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":45:75)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":46:20)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:25)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:36)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:4)
+#loc36 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":144:21)
+#loc37 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":145:23)
+#loc38 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":146:19)
+#loc39 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":146:7)
+#loc40 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":147:29)
+#loc41 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":148:29)
+#loc42 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:31)
+#loc43 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:27)
+#loc44 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:16)
+#loc45 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:27)
+#loc46 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:17)
+#loc48 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:31)
+#loc49 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:21)
+#loc50 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:12)
+#loc51 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:35)
+#loc52 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:69)
+#loc53 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:11)
+#loc54 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:4)
+#loc56 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":87:29)
+#loc57 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":87:11)
+#loc58 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":87:4)
+#loc60 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":65:30)
+#loc61 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":65:15)
+#loc62 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":65:11)
+#loc63 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":65:4)
+#loc64 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":118:0)
+#loc65 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":127:31)
+#loc66 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":127:11)
+#loc67 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":127:4)
+#loc69 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":165:42)
+#loc70 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":165:11)
+#loc71 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":165:4)
+#loc78 = loc("r0_numel"(#loc1))
+#loc79 = loc("xoffset"(#loc2))
+#loc80 = loc("xoffset"(#loc3))
+#loc81 = loc("xindex"(#loc4))
+#loc82 = loc("xindex"(#loc5))
+#loc83 = loc("xindex"(#loc6))
+#loc84 = loc("xmask"(#loc7))
+#loc85 = loc("r0_base"(#loc8))
+#loc86 = loc("r0_base"(#loc9))
+#loc87 = loc("x0"(#loc10))
+#loc88 = loc("x1"(#loc11))
+#loc89 = loc("_tmp2"(#loc12))
+#loc90 = loc("_tmp2_index"(#loc13))
+#loc91 = loc("_tmp2"(#loc14))
+#loc92 = loc("r0_index"(#loc15))
+#loc93 = loc("r0_mask"(#loc16))
+#loc94 = loc("tmp0"(#loc17))
+#loc95 = loc("tmp0"(#loc18))
+#loc96 = loc("tmp0"(#loc19))
+#loc97 = loc("tmp0"(#loc20))
+#loc98 = loc("tmp0"(#loc21))
+#loc99 = loc("tmp0"(#loc22))
+#loc100 = loc("tmp0"(#loc23))
+#loc101 = loc("_tmp2"(#loc25))
+#loc102 = loc("_tmp2"(#loc26))
+#loc103 = loc("_tmp2_index"(#loc27))
+#loc104 = loc("_tmp2_index"(#loc28))
+#loc105 = loc("tmp2"(#loc31))
+#loc110 = loc("mask"(#loc36))
+#loc111 = loc("equal"(#loc37))
+#loc112 = loc("a_isnan"(#loc40))
+#loc113 = loc("b_isnan"(#loc41))
+#loc114 = loc("mask"(#loc42))
+#loc115 = loc("mask"(#loc43))
+#loc116 = loc("mask"(#loc44))
+#loc117 = loc("equal"(#loc45))
+#loc118 = loc("equal"(#loc46))
+#loc119 = loc("mask"(#loc48))
+#loc120 = loc("mask"(#loc49))
+#loc121 = loc("mask"(#loc50))
+#loc126 = loc("_tmp2_index"(#loc91))
+#loc127 = loc("mask"(#loc110))
+#loc128 = loc("equal"(#loc111))
+#loc129 = loc("mask"(#loc116))
+#loc130 = loc("equal"(#loc118))

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ttgir

@@ -0,0 +1,218 @@
+#blocked = #ttg.blocked<{sizePerThread = [1, 1], threadsPerWarp = [8, 4], warpsPerCTA = [8, 1], order = [1, 0]}>
+#blocked1 = #ttg.blocked<{sizePerThread = [1, 1], threadsPerWarp = [32, 1], warpsPerCTA = [2, 4], order = [0, 1]}>
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":18:0)
+#loc1 = loc(unknown)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":45:75)
+#loc44 = loc("in_ptr0"(#loc))
+#loc45 = loc("out_ptr0"(#loc))
+#loc46 = loc("ks0"(#loc))
+#loc47 = loc("ks1"(#loc))
+#loc48 = loc("xnumel"(#loc))
+#loc49 = loc("r0_numel"(#loc))
+#loc85 = loc(callsite(#loc1 at #loc39))
+module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 8 : i32, ttg.target = "cuda:90", "ttg.threads-per-warp" = 32 : i32} {
+  tt.func public @triton_red_fused_argmax_1(%in_ptr0: !tt.ptr<f32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr0: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 {tt.divisibility = 16 : i32} loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<32000> : tensor<64x1xi64, #blocked> loc(#loc1)
+    %cst_0 = arith.constant dense<0.000000e+00> : tensor<64x4xf32, #blocked> loc(#loc1)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc1)
+    %c32000_i32 = arith.constant 32000 : i32 loc(#loc1)
+    %c4_i32 = arith.constant 4 : i32 loc(#loc1)
+    %cst_1 = arith.constant dense<true> : tensor<64x4xi1, #blocked> loc(#loc1)
+    %true = arith.constant true loc(#loc1)
+    %cst_2 = arith.constant dense<32000> : tensor<1x4xi32, #blocked> loc(#loc1)
+    %cst_3 = arith.constant dense<2147483647> : tensor<64x4xi32, #blocked> loc(#loc1)
+    %cst_4 = arith.constant dense<0xFF800000> : tensor<64x4xf32, #blocked> loc(#loc1)
+    %c64_i32 = arith.constant 64 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc50)
+    %xoffset_5 = arith.muli %xoffset, %c64_i32 : i32 loc(#loc51)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc52)
+    %xindex_6 = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked1}>> loc(#loc52)
+    %xindex_7 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<64x1xi32, #blocked> loc(#loc52)
+    %xindex_8 = tt.expand_dims %xindex_6 {axis = 1 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked1}>> -> tensor<64x1xi32, #blocked1> loc(#loc52)
+    %xindex_9 = tt.splat %xoffset_5 : i32 -> tensor<64x1xi32, #blocked> loc(#loc53)
+    %xindex_10 = tt.splat %xoffset_5 : i32 -> tensor<64x1xi32, #blocked1> loc(#loc53)
+    %xindex_11 = arith.addi %xindex_9, %xindex_7 : tensor<64x1xi32, #blocked> loc(#loc53)
+    %xindex_12 = arith.addi %xindex_10, %xindex_8 : tensor<64x1xi32, #blocked1> loc(#loc53)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32, #blocked> loc(#loc54)
+    %xmask_13 = tt.splat %xnumel : i32 -> tensor<64x1xi32, #blocked1> loc(#loc54)
+    %xmask_14 = arith.cmpi slt, %xindex_11, %xmask : tensor<64x1xi32, #blocked> loc(#loc54)
+    %xmask_15 = arith.cmpi slt, %xindex_12, %xmask_13 : tensor<64x1xi32, #blocked1> loc(#loc54)
+    %r0_base = tt.make_range {end = 4 : i32, start = 0 : i32} : tensor<4xi32, #ttg.slice<{dim = 0, parent = #blocked}>> loc(#loc55)
+    %r0_base_16 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<4xi32, #ttg.slice<{dim = 0, parent = #blocked}>> -> tensor<1x4xi32, #blocked> loc(#loc55)
+    %x0 = arith.extsi %xindex_11 : tensor<64x1xi32, #blocked> to tensor<64x1xi64, #blocked> loc(#loc56)
+    %x0_17 = tt.splat %ks0 : i64 -> tensor<64x1xi64, #blocked> loc(#loc56)
+    %x0_18 = arith.remsi %x0, %x0_17 : tensor<64x1xi64, #blocked> loc(#loc56)
+    %x1 = arith.divsi %x0, %x0_17 : tensor<64x1xi64, #blocked> loc(#loc57)
+    %tmp0 = arith.muli %x0_18, %cst : tensor<64x1xi64, #blocked> loc(#loc58)
+    %tmp0_19 = tt.broadcast %tmp0 : tensor<64x1xi64, #blocked> -> tensor<64x4xi64, #blocked> loc(#loc59)
+    %tmp0_20 = tt.splat %ks1 : i64 -> tensor<64x1xi64, #blocked> loc(#loc60)
+    %tmp0_21 = arith.muli %tmp0_20, %x1 : tensor<64x1xi64, #blocked> loc(#loc60)
+    %tmp0_22 = tt.broadcast %tmp0_21 : tensor<64x1xi64, #blocked> -> tensor<64x4xi64, #blocked> loc(#loc61)
+    %tmp0_23 = tt.splat %in_ptr0 : !tt.ptr<f32> -> tensor<64x4x!tt.ptr<f32>, #blocked> loc(#loc62)
+    %tmp0_24 = tt.broadcast %xmask_14 : tensor<64x1xi1, #blocked> -> tensor<64x4xi1, #blocked> loc(#loc63)
+    %_tmp2_index:2 = scf.for %r0_offset = %c0_i32 to %c32000_i32 step %c4_i32 iter_args(%_tmp2 = %cst_4, %_tmp2_index_25 = %cst_3) -> (tensor<64x4xf32, #blocked>, tensor<64x4xi32, #blocked>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x4xi32, #blocked> loc(#loc65)
+      %r0_index_26 = arith.addi %r0_index, %r0_base_16 : tensor<1x4xi32, #blocked> loc(#loc65)
+      %r0_mask = arith.cmpi slt, %r0_index_26, %cst_2 : tensor<1x4xi32, #blocked> loc(#loc66)
+      %tmp0_27 = arith.extsi %r0_index_26 : tensor<1x4xi32, #blocked> to tensor<1x4xi64, #blocked> loc(#loc59)
+      %tmp0_28 = tt.broadcast %tmp0_27 : tensor<1x4xi64, #blocked> -> tensor<64x4xi64, #blocked> loc(#loc59)
+      %tmp0_29 = arith.addi %tmp0_28, %tmp0_19 : tensor<64x4xi64, #blocked> loc(#loc59)
+      %tmp0_30 = arith.addi %tmp0_29, %tmp0_22 : tensor<64x4xi64, #blocked> loc(#loc61)
+      %tmp0_31 = tt.addptr %tmp0_23, %tmp0_30 : tensor<64x4x!tt.ptr<f32>, #blocked>, tensor<64x4xi64, #blocked> loc(#loc62)
+      %tmp0_32 = tt.broadcast %r0_mask : tensor<1x4xi1, #blocked> -> tensor<64x4xi1, #blocked> loc(#loc63)
+      %tmp0_33 = arith.andi %tmp0_32, %tmp0_24 : tensor<64x4xi1, #blocked> loc(#loc63)
+      %tmp0_34 = tt.load %tmp0_31, %tmp0_33, %cst_0 evictionPolicy = evict_first : tensor<64x4x!tt.ptr<f32>, #blocked> loc(#loc67)
+      %mask = arith.cmpf ogt, %_tmp2, %tmp0_34 : tensor<64x4xf32, #blocked> loc(#loc110)
+      %equal = arith.cmpf oeq, %_tmp2, %tmp0_34 : tensor<64x4xf32, #blocked> loc(#loc111)
+      %a_isnan = arith.cmpf une, %_tmp2, %_tmp2 : tensor<64x4xf32, #blocked> loc(#loc90)
+      %b_isnan = arith.cmpf une, %tmp0_34, %tmp0_34 : tensor<64x4xf32, #blocked> loc(#loc91)
+      %mask_35 = arith.xori %b_isnan, %cst_1 : tensor<64x4xi1, #blocked> loc(#loc92)
+      %mask_36 = arith.andi %a_isnan, %mask_35 : tensor<64x4xi1, #blocked> loc(#loc93)
+      %mask_37 = arith.ori %mask, %mask_36 : tensor<64x4xi1, #blocked> loc(#loc112)
+      %equal_38 = arith.andi %a_isnan, %b_isnan : tensor<64x4xi1, #blocked> loc(#loc95)
+      %equal_39 = arith.ori %equal, %equal_38 : tensor<64x4xi1, #blocked> loc(#loc113)
+      %mask_40 = tt.broadcast %r0_index_26 : tensor<1x4xi32, #blocked> -> tensor<64x4xi32, #blocked> loc(#loc97)
+      %mask_41 = arith.cmpi slt, %_tmp2_index_25, %mask_40 : tensor<64x4xi32, #blocked> loc(#loc97)
+      %mask_42 = arith.andi %equal_39, %mask_41 : tensor<64x4xi1, #blocked> loc(#loc98)
+      %mask_43 = arith.ori %mask_37, %mask_42 : tensor<64x4xi1, #blocked> loc(#loc99)
+      %5 = arith.select %mask_43, %_tmp2, %tmp0_34 : tensor<64x4xi1, #blocked>, tensor<64x4xf32, #blocked> loc(#loc80)
+      %6 = arith.select %mask_43, %_tmp2_index_25, %mask_40 : tensor<64x4xi1, #blocked>, tensor<64x4xi32, #blocked> loc(#loc81)
+      %_tmp2_44 = arith.select %tmp0_33, %5, %_tmp2 : tensor<64x4xi1, #blocked>, tensor<64x4xf32, #blocked> loc(#loc82)
+      %_tmp2_index_45 = arith.select %tmp0_33, %6, %_tmp2_index_25 : tensor<64x4xi1, #blocked>, tensor<64x4xi32, #blocked> loc(#loc83)
+      scf.yield %_tmp2_44, %_tmp2_index_45 : tensor<64x4xf32, #blocked>, tensor<64x4xi32, #blocked> loc(#loc37)
+    } loc(#loc87)
+    %0:2 = "tt.reduce"(%_tmp2_index#0, %_tmp2_index#1) <{axis = 1 : i32}> ({
+    ^bb0(%arg6: f32 loc(callsite(#loc1 at #loc39)), %arg7: i32 loc(callsite(#loc1 at #loc39)), %arg8: f32 loc(callsite(#loc1 at #loc39)), %arg9: i32 loc(callsite(#loc1 at #loc39))):
+      %mask = arith.cmpf ogt, %arg6, %arg8 : f32 loc(#loc114)
+      %equal = arith.cmpf oeq, %arg6, %arg8 : f32 loc(#loc115)
+      %a_isnan = arith.cmpf une, %arg6, %arg6 : f32 loc(#loc100)
+      %b_isnan = arith.cmpf une, %arg8, %arg8 : f32 loc(#loc101)
+      %mask_25 = arith.xori %b_isnan, %true : i1 loc(#loc102)
+      %mask_26 = arith.andi %a_isnan, %mask_25 : i1 loc(#loc103)
+      %mask_27 = arith.ori %mask, %mask_26 : i1 loc(#loc116)
+      %equal_28 = arith.andi %a_isnan, %b_isnan : i1 loc(#loc104)
+      %equal_29 = arith.ori %equal, %equal_28 : i1 loc(#loc117)
+      %mask_30 = arith.cmpi slt, %arg7, %arg9 : i32 loc(#loc105)
+      %mask_31 = arith.andi %equal_29, %mask_30 : i1 loc(#loc106)
+      %mask_32 = arith.ori %mask_27, %mask_31 : i1 loc(#loc107)
+      %5 = arith.select %mask_32, %arg6, %arg8 : f32 loc(#loc108)
+      %6 = arith.select %mask_32, %arg7, %arg9 : i32 loc(#loc109)
+      tt.reduce.return %5, %6 : f32, i32 loc(#loc84)
+    }) : (tensor<64x4xf32, #blocked>, tensor<64x4xi32, #blocked>) -> (tensor<64xf32, #ttg.slice<{dim = 1, parent = #blocked}>>, tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>>) loc(#loc84)
+    %tmp2 = tt.expand_dims %0#1 {axis = 1 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<64x1xi32, #blocked> loc(#loc86)
+    %1 = tt.splat %out_ptr0 : !tt.ptr<i64> -> tensor<64x1x!tt.ptr<i64>, #blocked1> loc(#loc41)
+    %2 = tt.addptr %1, %xindex_12 : tensor<64x1x!tt.ptr<i64>, #blocked1>, tensor<64x1xi32, #blocked1> loc(#loc41)
+    %3 = ttg.convert_layout %tmp2 : tensor<64x1xi32, #blocked> -> tensor<64x1xi32, #blocked1> loc(#loc42)
+    %4 = arith.extsi %3 : tensor<64x1xi32, #blocked1> to tensor<64x1xi64, #blocked1> loc(#loc42)
+    tt.store %2, %4, %xmask_15 : tensor<64x1x!tt.ptr<i64>, #blocked1> loc(#loc42)
+    tt.return loc(#loc43)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":24:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":25:37)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":27:19)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":28:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:47)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:41)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:56)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:52)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:34)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:71)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":32:40)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":33:31)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":34:29)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:61)
+#loc20 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":144:21)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":41:38)
+#loc22 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":145:23)
+#loc23 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":147:29)
+#loc24 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":148:29)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:31)
+#loc26 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:27)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:16)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:27)
+#loc29 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:17)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:31)
+#loc31 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:21)
+#loc32 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:12)
+#loc33 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:35)
+#loc34 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:69)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":43:54)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:66)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:8)
+#loc38 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":165:42)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":46:20)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:25)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:36)
+#loc43 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:4)
+#loc50 = loc("xoffset"(#loc2))
+#loc51 = loc("xoffset"(#loc3))
+#loc52 = loc("xindex"(#loc4))
+#loc53 = loc("xindex"(#loc5))
+#loc54 = loc("xmask"(#loc6))
+#loc55 = loc("r0_base"(#loc7))
+#loc56 = loc("x0"(#loc8))
+#loc57 = loc("x1"(#loc9))
+#loc58 = loc("tmp0"(#loc10))
+#loc59 = loc("tmp0"(#loc11))
+#loc60 = loc("tmp0"(#loc12))
+#loc61 = loc("tmp0"(#loc13))
+#loc62 = loc("tmp0"(#loc14))
+#loc63 = loc("tmp0"(#loc15))
+#loc64 = loc("_tmp2"(#loc16))
+#loc65 = loc("r0_index"(#loc17))
+#loc66 = loc("r0_mask"(#loc18))
+#loc67 = loc("tmp0"(#loc19))
+#loc68 = loc("mask"(#loc20))
+#loc69 = loc("equal"(#loc22))
+#loc70 = loc("a_isnan"(#loc23))
+#loc71 = loc("b_isnan"(#loc24))
+#loc72 = loc("mask"(#loc25))
+#loc73 = loc("mask"(#loc26))
+#loc74 = loc("mask"(#loc27))
+#loc75 = loc("equal"(#loc28))
+#loc76 = loc("equal"(#loc29))
+#loc77 = loc("mask"(#loc30))
+#loc78 = loc("mask"(#loc31))
+#loc79 = loc("mask"(#loc32))
+#loc80 = loc(callsite(#loc33 at #loc21))
+#loc81 = loc(callsite(#loc34 at #loc21))
+#loc82 = loc("_tmp2"(#loc35))
+#loc83 = loc("_tmp2_index"(#loc36))
+#loc84 = loc(callsite(#loc38 at #loc39))
+#loc86 = loc("tmp2"(#loc40))
+#loc87 = loc("_tmp2_index"(#loc64))
+#loc88 = loc("mask"(#loc68))
+#loc89 = loc("equal"(#loc69))
+#loc90 = loc(callsite(#loc70 at #loc21))
+#loc91 = loc(callsite(#loc71 at #loc21))
+#loc92 = loc(callsite(#loc72 at #loc21))
+#loc93 = loc(callsite(#loc73 at #loc21))
+#loc94 = loc("mask"(#loc74))
+#loc95 = loc(callsite(#loc75 at #loc21))
+#loc96 = loc("equal"(#loc76))
+#loc97 = loc(callsite(#loc77 at #loc21))
+#loc98 = loc(callsite(#loc78 at #loc21))
+#loc99 = loc(callsite(#loc79 at #loc21))
+#loc100 = loc(callsite(#loc70 at #loc84))
+#loc101 = loc(callsite(#loc71 at #loc84))
+#loc102 = loc(callsite(#loc72 at #loc84))
+#loc103 = loc(callsite(#loc73 at #loc84))
+#loc104 = loc(callsite(#loc75 at #loc84))
+#loc105 = loc(callsite(#loc77 at #loc84))
+#loc106 = loc(callsite(#loc78 at #loc84))
+#loc107 = loc(callsite(#loc79 at #loc84))
+#loc108 = loc(callsite(#loc33 at #loc84))
+#loc109 = loc(callsite(#loc34 at #loc84))
+#loc110 = loc(callsite(#loc88 at #loc21))
+#loc111 = loc(callsite(#loc89 at #loc21))
+#loc112 = loc(callsite(#loc94 at #loc21))
+#loc113 = loc(callsite(#loc96 at #loc21))
+#loc114 = loc(callsite(#loc88 at #loc84))
+#loc115 = loc(callsite(#loc89 at #loc84))
+#loc116 = loc(callsite(#loc94 at #loc84))
+#loc117 = loc(callsite(#loc96 at #loc84))

SpecForge-ext/cache/compiled_kernels/triton/0/25SMJXR2INGZCZI64NAKGLW77JZOIG6LAES6NHHOQFOTKNXS6PHA/triton_red_fused_argmax_1.ttir

@@ -0,0 +1,217 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":18:0)
+#loc1 = loc(unknown)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":45:75)
+#loc48 = loc("in_ptr0"(#loc))
+#loc49 = loc("out_ptr0"(#loc))
+#loc50 = loc("ks0"(#loc))
+#loc51 = loc("ks1"(#loc))
+#loc52 = loc("xnumel"(#loc))
+#loc53 = loc("r0_numel"(#loc))
+#loc54 = loc(callsite(#loc1 at #loc2))
+module {
+  tt.func public @triton_red_fused_argmax_1(%in_ptr0: !tt.ptr<f32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr0: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 {tt.divisibility = 16 : i32} loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %true = arith.constant true loc(#loc54)
+    %cst = arith.constant dense<true> : tensor<64x4xi1> loc(#loc1)
+    %c4_i32 = arith.constant 4 : i32 loc(#loc3)
+    %c32000_i32 = arith.constant 32000 : i32 loc(#loc3)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc3)
+    %cst_0 = arith.constant dense<0.000000e+00> : tensor<64x4xf32> loc(#loc1)
+    %cst_1 = arith.constant dense<32000> : tensor<64x1xi64> loc(#loc1)
+    %cst_2 = arith.constant dense<32000> : tensor<1x4xi32> loc(#loc1)
+    %_tmp2_index = arith.constant dense<2147483647> : tensor<64x4xi32> loc(#loc55)
+    %_tmp2 = arith.constant dense<0xFF800000> : tensor<64x4xf32> loc(#loc56)
+    %c64_i32 = arith.constant 64 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc57)
+    %xoffset_3 = arith.muli %xoffset, %c64_i32 : i32 loc(#loc58)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32> loc(#loc59)
+    %xindex_4 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc60)
+    %xindex_5 = tt.splat %xoffset_3 : i32 -> tensor<64x1xi32> loc(#loc61)
+    %xindex_6 = arith.addi %xindex_5, %xindex_4 : tensor<64x1xi32> loc(#loc61)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32> loc(#loc62)
+    %xmask_7 = arith.cmpi slt, %xindex_6, %xmask : tensor<64x1xi32> loc(#loc62)
+    %r0_base = tt.make_range {end = 4 : i32, start = 0 : i32} : tensor<4xi32> loc(#loc63)
+    %r0_base_8 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<4xi32> -> tensor<1x4xi32> loc(#loc64)
+    %x0 = arith.extsi %xindex_6 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc65)
+    %x0_9 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc65)
+    %x0_10 = arith.remsi %x0, %x0_9 : tensor<64x1xi64> loc(#loc65)
+    %x1 = arith.divsi %x0, %x0_9 : tensor<64x1xi64> loc(#loc66)
+    %_tmp2_index_11:2 = scf.for %r0_offset = %c0_i32 to %c32000_i32 step %c4_i32 iter_args(%_tmp2_12 = %_tmp2, %_tmp2_index_13 = %_tmp2_index) -> (tensor<64x4xf32>, tensor<64x4xi32>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x4xi32> loc(#loc68)
+      %r0_index_14 = arith.addi %r0_index, %r0_base_8 : tensor<1x4xi32> loc(#loc68)
+      %r0_mask = arith.cmpi slt, %r0_index_14, %cst_2 : tensor<1x4xi32> loc(#loc69)
+      %tmp0 = arith.muli %x0_10, %cst_1 : tensor<64x1xi64> loc(#loc70)
+      %tmp0_15 = arith.extsi %r0_index_14 : tensor<1x4xi32> to tensor<1x4xi64> loc(#loc71)
+      %tmp0_16 = tt.broadcast %tmp0_15 : tensor<1x4xi64> -> tensor<64x4xi64> loc(#loc71)
+      %tmp0_17 = tt.broadcast %tmp0 : tensor<64x1xi64> -> tensor<64x4xi64> loc(#loc71)
+      %tmp0_18 = arith.addi %tmp0_16, %tmp0_17 : tensor<64x4xi64> loc(#loc71)
+      %tmp0_19 = tt.splat %ks1 : i64 -> tensor<64x1xi64> loc(#loc72)
+      %tmp0_20 = arith.muli %tmp0_19, %x1 : tensor<64x1xi64> loc(#loc72)
+      %tmp0_21 = tt.broadcast %tmp0_20 : tensor<64x1xi64> -> tensor<64x4xi64> loc(#loc73)
+      %tmp0_22 = arith.addi %tmp0_18, %tmp0_21 : tensor<64x4xi64> loc(#loc73)
+      %tmp0_23 = tt.splat %in_ptr0 : !tt.ptr<f32> -> tensor<64x4x!tt.ptr<f32>> loc(#loc74)
+      %tmp0_24 = tt.addptr %tmp0_23, %tmp0_22 : tensor<64x4x!tt.ptr<f32>>, tensor<64x4xi64> loc(#loc74)
+      %tmp0_25 = tt.broadcast %r0_mask : tensor<1x4xi1> -> tensor<64x4xi1> loc(#loc75)
+      %tmp0_26 = tt.broadcast %xmask_7 : tensor<64x1xi1> -> tensor<64x4xi1> loc(#loc75)
+      %tmp0_27 = arith.andi %tmp0_25, %tmp0_26 : tensor<64x4xi1> loc(#loc75)
+      %tmp0_28 = tt.load %tmp0_24, %tmp0_27, %cst_0 evictionPolicy = evict_first : tensor<64x4x!tt.ptr<f32>> loc(#loc76)
+      %mask = arith.cmpf ogt, %_tmp2_12, %tmp0_28 : tensor<64x4xf32> loc(#loc118)
+      %equal = arith.cmpf oeq, %_tmp2_12, %tmp0_28 : tensor<64x4xf32> loc(#loc119)
+      %a_isnan = arith.cmpf une, %_tmp2_12, %_tmp2_12 : tensor<64x4xf32> loc(#loc98)
+      %b_isnan = arith.cmpf une, %tmp0_28, %tmp0_28 : tensor<64x4xf32> loc(#loc99)
+      %mask_29 = arith.xori %b_isnan, %cst : tensor<64x4xi1> loc(#loc100)
+      %mask_30 = arith.andi %a_isnan, %mask_29 : tensor<64x4xi1> loc(#loc101)
+      %mask_31 = arith.ori %mask, %mask_30 : tensor<64x4xi1> loc(#loc120)
+      %equal_32 = arith.andi %a_isnan, %b_isnan : tensor<64x4xi1> loc(#loc103)
+      %equal_33 = arith.ori %equal, %equal_32 : tensor<64x4xi1> loc(#loc121)
+      %mask_34 = tt.broadcast %r0_index_14 : tensor<1x4xi32> -> tensor<64x4xi32> loc(#loc105)
+      %mask_35 = arith.cmpi slt, %_tmp2_index_13, %mask_34 : tensor<64x4xi32> loc(#loc105)
+      %mask_36 = arith.andi %equal_33, %mask_35 : tensor<64x4xi1> loc(#loc106)
+      %mask_37 = arith.ori %mask_31, %mask_36 : tensor<64x4xi1> loc(#loc107)
+      %4 = arith.select %mask_37, %_tmp2_12, %tmp0_28 : tensor<64x4xi1>, tensor<64x4xf32> loc(#loc89)
+      %5 = arith.select %mask_37, %_tmp2_index_13, %mask_34 : tensor<64x4xi1>, tensor<64x4xi32> loc(#loc90)
+      %_tmp2_38 = arith.select %tmp0_27, %4, %_tmp2_12 : tensor<64x4xi1>, tensor<64x4xf32> loc(#loc91)
+      %_tmp2_index_39 = arith.select %tmp0_27, %5, %_tmp2_index_13 : tensor<64x4xi1>, tensor<64x4xi32> loc(#loc92)
+      scf.yield %_tmp2_38, %_tmp2_index_39 : tensor<64x4xf32>, tensor<64x4xi32> loc(#loc42)
+    } loc(#loc95)
+    %0:2 = "tt.reduce"(%_tmp2_index_11#0, %_tmp2_index_11#1) <{axis = 1 : i32}> ({
+    ^bb0(%arg6: f32 loc(callsite(#loc1 at #loc2)), %arg7: i32 loc(callsite(#loc1 at #loc2)), %arg8: f32 loc(callsite(#loc1 at #loc2)), %arg9: i32 loc(callsite(#loc1 at #loc2))):
+      %mask = arith.cmpf ogt, %arg6, %arg8 : f32 loc(#loc122)
+      %equal = arith.cmpf oeq, %arg6, %arg8 : f32 loc(#loc123)
+      %a_isnan = arith.cmpf une, %arg6, %arg6 : f32 loc(#loc108)
+      %b_isnan = arith.cmpf une, %arg8, %arg8 : f32 loc(#loc109)
+      %mask_12 = arith.xori %b_isnan, %true : i1 loc(#loc110)
+      %mask_13 = arith.andi %a_isnan, %mask_12 : i1 loc(#loc111)
+      %mask_14 = arith.ori %mask, %mask_13 : i1 loc(#loc124)
+      %equal_15 = arith.andi %a_isnan, %b_isnan : i1 loc(#loc112)
+      %equal_16 = arith.ori %equal, %equal_15 : i1 loc(#loc125)
+      %mask_17 = arith.cmpi slt, %arg7, %arg9 : i32 loc(#loc113)
+      %mask_18 = arith.andi %equal_16, %mask_17 : i1 loc(#loc114)
+      %mask_19 = arith.ori %mask_14, %mask_18 : i1 loc(#loc115)
+      %4 = arith.select %mask_19, %arg6, %arg8 : f32 loc(#loc116)
+      %5 = arith.select %mask_19, %arg7, %arg9 : i32 loc(#loc117)
+      tt.reduce.return %4, %5 : f32, i32 loc(#loc93)
+    }) : (tensor<64x4xf32>, tensor<64x4xi32>) -> (tensor<64xf32>, tensor<64xi32>) loc(#loc93)
+    %tmp2 = tt.expand_dims %0#1 {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc94)
+    %1 = tt.splat %out_ptr0 : !tt.ptr<i64> -> tensor<64x1x!tt.ptr<i64>> loc(#loc45)
+    %2 = tt.addptr %1, %xindex_6 : tensor<64x1x!tt.ptr<i64>>, tensor<64x1xi32> loc(#loc45)
+    %3 = arith.extsi %tmp2 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc46)
+    tt.store %2, %3, %xmask_7 : tensor<64x1x!tt.ptr<i64>> loc(#loc46)
+    tt.return loc(#loc47)
+  } loc(#loc)
+} loc(#loc)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":32:40)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":30:58)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":29:55)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:28)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":22:33)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:36)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:44)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":23:23)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":24:21)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":25:27)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":25:37)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":27:19)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":28:19)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":33:31)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":34:29)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:47)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:41)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:56)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:52)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:34)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:71)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":38:61)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":144:21)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":41:38)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":145:23)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":147:29)
+#loc29 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":148:29)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:31)
+#loc31 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:27)
+#loc32 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":149:16)
+#loc33 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:27)
+#loc34 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":151:17)
+#loc35 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:31)
+#loc36 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:21)
+#loc37 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":154:12)
+#loc38 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:35)
+#loc39 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":155:69)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":43:54)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:66)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":44:8)
+#loc43 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":165:42)
+#loc44 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":46:20)
+#loc45 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:25)
+#loc46 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:36)
+#loc47 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/eu/ceui6qrb2t3lmzs3ljrqtcomt4b2q6svzo24j6mmryaiovr6kp7y.py":47:4)
+#loc55 = loc("_tmp2_index"(#loc4))
+#loc56 = loc("_tmp2"(#loc5))
+#loc57 = loc("xoffset"(#loc6))
+#loc58 = loc("xoffset"(#loc7))
+#loc59 = loc("xindex"(#loc8))
+#loc60 = loc("xindex"(#loc9))
+#loc61 = loc("xindex"(#loc10))
+#loc62 = loc("xmask"(#loc11))
+#loc63 = loc("r0_base"(#loc12))
+#loc64 = loc("r0_base"(#loc13))
+#loc65 = loc("x0"(#loc14))
+#loc66 = loc("x1"(#loc15))
+#loc67 = loc("_tmp2"(#loc3))
+#loc68 = loc("r0_index"(#loc16))
+#loc69 = loc("r0_mask"(#loc17))
+#loc70 = loc("tmp0"(#loc18))
+#loc71 = loc("tmp0"(#loc19))
+#loc72 = loc("tmp0"(#loc20))
+#loc73 = loc("tmp0"(#loc21))
+#loc74 = loc("tmp0"(#loc22))
+#loc75 = loc("tmp0"(#loc23))
+#loc76 = loc("tmp0"(#loc24))
+#loc77 = loc("mask"(#loc25))
+#loc78 = loc("equal"(#loc27))
+#loc79 = loc("a_isnan"(#loc28))
+#loc80 = loc("b_isnan"(#loc29))
+#loc81 = loc("mask"(#loc30))
+#loc82 = loc("mask"(#loc31))
+#loc83 = loc("mask"(#loc32))
+#loc84 = loc("equal"(#loc33))
+#loc85 = loc("equal"(#loc34))
+#loc86 = loc("mask"(#loc35))
+#loc87 = loc("mask"(#loc36))
+#loc88 = loc("mask"(#loc37))
+#loc89 = loc(callsite(#loc38 at #loc26))
+#loc90 = loc(callsite(#loc39 at #loc26))
+#loc91 = loc("_tmp2"(#loc40))
+#loc92 = loc("_tmp2_index"(#loc41))
+#loc93 = loc(callsite(#loc43 at #loc2))
+#loc94 = loc("tmp2"(#loc44))
+#loc95 = loc("_tmp2_index"(#loc67))
+#loc96 = loc("mask"(#loc77))
+#loc97 = loc("equal"(#loc78))
+#loc98 = loc(callsite(#loc79 at #loc26))
+#loc99 = loc(callsite(#loc80 at #loc26))
+#loc100 = loc(callsite(#loc81 at #loc26))
+#loc101 = loc(callsite(#loc82 at #loc26))
+#loc102 = loc("mask"(#loc83))
+#loc103 = loc(callsite(#loc84 at #loc26))
+#loc104 = loc("equal"(#loc85))
+#loc105 = loc(callsite(#loc86 at #loc26))
+#loc106 = loc(callsite(#loc87 at #loc26))
+#loc107 = loc(callsite(#loc88 at #loc26))
+#loc108 = loc(callsite(#loc79 at #loc93))
+#loc109 = loc(callsite(#loc80 at #loc93))
+#loc110 = loc(callsite(#loc81 at #loc93))
+#loc111 = loc(callsite(#loc82 at #loc93))
+#loc112 = loc(callsite(#loc84 at #loc93))
+#loc113 = loc(callsite(#loc86 at #loc93))
+#loc114 = loc(callsite(#loc87 at #loc93))
+#loc115 = loc(callsite(#loc88 at #loc93))
+#loc116 = loc(callsite(#loc38 at #loc93))
+#loc117 = loc(callsite(#loc39 at #loc93))
+#loc118 = loc(callsite(#loc96 at #loc26))
+#loc119 = loc(callsite(#loc97 at #loc26))
+#loc120 = loc(callsite(#loc102 at #loc26))
+#loc121 = loc(callsite(#loc104 at #loc26))
+#loc122 = loc(callsite(#loc96 at #loc93))
+#loc123 = loc(callsite(#loc97 at #loc93))
+#loc124 = loc(callsite(#loc102 at #loc93))
+#loc125 = loc(callsite(#loc104 at #loc93))

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/__grp__triton_red_fused__to_copy_clone_slice_sum_transpose_5.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_red_fused__to_copy_clone_slice_sum_transpose_5.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.source", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.cubin", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.json

@@ -0,0 +1 @@
+{"hash": "d1c2e6527ce27b628a96c1a250025b39aad19d679fce295e820390aa7ae64b66", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 4, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 1024, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_red_fused__to_copy_clone_slice_sum_transpose_5"}

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir

@@ -0,0 +1,204 @@
+; ModuleID = 'LLVMDialectModule'
+source_filename = "LLVMDialectModule"
+target datalayout = "e-p3:32:32-p4:32:32-p5:32:32-p6:32:32-p7:32:32-i64:64-i128:128-v16:16-v32:32-n16:32:64"
+
+@global_smem = external addrspace(3) global [0 x i8], align 16
+
+; Function Attrs: nounwind
+define ptx_kernel void @triton_red_fused__to_copy_clone_slice_sum_transpose_5(ptr addrspace(1) %0, ptr addrspace(1) %1, i64 %2, i64 %3, i32 %4, i32 %5, ptr addrspace(1) readnone captures(none) %6, ptr addrspace(1) readnone captures(none) %7) local_unnamed_addr #0 !dbg !4 {
+  %9 = tail call i32 @llvm.nvvm.read.ptx.sreg.ctaid.x(), !dbg !7
+  %10 = shl i32 %9, 5, !dbg !8
+  %11 = tail call i32 @llvm.nvvm.read.ptx.sreg.tid.x(), !dbg !9
+  %12 = and i32 %11, 31, !dbg !9
+  %13 = or disjoint i32 %10, %12, !dbg !10
+  %14 = icmp slt i32 %13, %4, !dbg !11
+  %15 = lshr i32 %11, 5, !dbg !12
+  %16 = and i32 %15, 3, !dbg !12
+  %17 = sext i32 %13 to i64, !dbg !13
+  %.frozen = freeze i64 %2, !dbg !14
+  %18 = sdiv i64 %17, %.frozen, !dbg !14
+  %19 = mul i64 %18, %.frozen, !dbg !13
+  %.decomposed = sub i64 %17, %19, !dbg !13
+  %20 = icmp sgt i32 %5, 0, !dbg !15
+  br i1 %20, label %.lr.ph, label %._crit_edge, !dbg !15
+
+.lr.ph:                                           ; preds = %8
+  %21 = mul i64 %3, %2, !dbg !16
+  %22 = mul i64 %21, %18, !dbg !17
+  %23 = getelementptr i32, ptr addrspace(1) %0, i64 %.decomposed
+  %invariant.gep = getelementptr i32, ptr addrspace(1) %23, i64 %22, !dbg !15
+  %24 = insertelement <4 x i1> poison, i1 %14, i64 0, !dbg !18
+  %25 = shufflevector <4 x i1> %24, <4 x i1> poison, <4 x i32> zeroinitializer, !dbg !18
+  %26 = insertelement <4 x i32> poison, i32 %5, i64 0, !dbg !19
+  %27 = shufflevector <4 x i32> %26, <4 x i32> poison, <4 x i32> zeroinitializer, !dbg !19
+  br label %28, !dbg !15
+
+28:                                               ; preds = %.lr.ph, %28
+  %29 = phi i32 [ 0, %.lr.ph ], [ %68, %28 ]
+  %30 = phi <4 x i64> [ zeroinitializer, %.lr.ph ], [ %67, %28 ]
+  %31 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %32 = or disjoint i32 %29, %16, !dbg !21
+  %33 = or disjoint i32 %32, 4, !dbg !21
+  %34 = or disjoint i32 %32, 8, !dbg !21
+  %35 = or disjoint i32 %32, 12, !dbg !21
+  %36 = insertelement <4 x i32> poison, i32 %32, i64 0, !dbg !19
+  %37 = insertelement <4 x i32> %36, i32 %33, i64 1, !dbg !19
+  %38 = insertelement <4 x i32> %37, i32 %34, i64 2, !dbg !19
+  %39 = insertelement <4 x i32> %38, i32 %35, i64 3, !dbg !19
+  %40 = icmp slt <4 x i32> %39, %27, !dbg !19
+  %41 = sext i32 %32 to i64, !dbg !22
+  %42 = sext i32 %33 to i64, !dbg !22
+  %43 = sext i32 %34 to i64, !dbg !22
+  %44 = sext i32 %35 to i64, !dbg !22
+  %45 = mul i64 %2, %41, !dbg !22
+  %46 = mul i64 %2, %42, !dbg !22
+  %47 = mul i64 %2, %43, !dbg !22
+  %48 = mul i64 %2, %44, !dbg !22
+  %gep = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %45, !dbg !23
+  %gep4 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %46, !dbg !23
+  %gep6 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %47, !dbg !23
+  %gep8 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %48, !dbg !23
+  %49 = and <4 x i1> %25, %40, !dbg !18
+  %50 = extractelement <4 x i1> %49, i64 0, !dbg !20
+  %51 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep, i64 %31, i1 %50) #5, !dbg !20
+  %52 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %53 = extractelement <4 x i1> %49, i64 1, !dbg !20
+  %54 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep4, i64 %52, i1 %53) #5, !dbg !20
+  %55 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %56 = extractelement <4 x i1> %49, i64 2, !dbg !20
+  %57 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep6, i64 %55, i1 %56) #5, !dbg !20
+  %58 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %59 = extractelement <4 x i1> %49, i64 3, !dbg !20
+  %60 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep8, i64 %58, i1 %59) #5, !dbg !20
+  %61 = insertelement <4 x i32> poison, i32 %51, i64 0, !dbg !24
+  %62 = insertelement <4 x i32> %61, i32 %54, i64 1, !dbg !24
+  %63 = insertelement <4 x i32> %62, i32 %57, i64 2, !dbg !24
+  %64 = insertelement <4 x i32> %63, i32 %60, i64 3, !dbg !24
+  %65 = sext <4 x i32> %64 to <4 x i64>, !dbg !24
+  %66 = select <4 x i1> %49, <4 x i64> %65, <4 x i64> zeroinitializer, !dbg !25
+  %67 = add <4 x i64> %66, %30, !dbg !25
+  %68 = add i32 %29, 16, !dbg !15
+  %69 = icmp slt i32 %68, %5, !dbg !15
+  br i1 %69, label %28, label %._crit_edge.loopexit, !dbg !15
+
+._crit_edge.loopexit:                             ; preds = %28
+  %70 = tail call i64 @llvm.vector.reduce.add.v4i64(<4 x i64> %67), !dbg !26
+  br label %._crit_edge, !dbg !26
+
+._crit_edge:                                      ; preds = %._crit_edge.loopexit, %8
+  %71 = phi i64 [ 0, %8 ], [ %70, %._crit_edge.loopexit ], !dbg !26
+  %.idx = shl nuw nsw i32 %12, 5, !dbg !30
+  %72 = getelementptr i8, ptr addrspace(3) @global_smem, i32 %.idx, !dbg !30
+  %73 = getelementptr i64, ptr addrspace(3) %72, i32 %16, !dbg !30
+  %74 = insertelement <1 x i64> poison, i64 %71, i64 0, !dbg !30
+  tail call void asm sideeffect "@$2 st.shared.b64 [ $0 + 0 ], $1;", "r,l,b"(ptr addrspace(3) %73, <1 x i64> %74, i1 true) #5, !dbg !30
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !30
+  %75 = icmp samesign ult i32 %11, 128, !dbg !30
+  %76 = getelementptr i64, ptr addrspace(3) @global_smem, i32 %11, !dbg !30
+  %77 = tail call i64 asm sideeffect "@$2 ld.shared.b64 $0, [ $1 + 0 ];", "=l,r,b"(ptr addrspace(3) %76, i1 %75) #5, !dbg !30
+  %extelt.offset = lshr i64 %77, 32, !dbg !30
+  %78 = trunc nuw i64 %extelt.offset to i32, !dbg !30
+  %79 = trunc i64 %77 to i32, !dbg !30
+  %80 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %79, i32 2, i32 31), !dbg !30
+  %81 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %78, i32 2, i32 31), !dbg !30
+  %82 = insertelement <2 x i32> poison, i32 %80, i64 0, !dbg !30
+  %83 = insertelement <2 x i32> %82, i32 %81, i64 1, !dbg !30
+  %84 = bitcast <2 x i32> %83 to i64, !dbg !30
+  %85 = add i64 %77, %84, !dbg !26
+  %extelt.offset2 = lshr i64 %85, 32, !dbg !30
+  %86 = trunc nuw i64 %extelt.offset2 to i32, !dbg !30
+  %87 = trunc i64 %85 to i32, !dbg !30
+  %88 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %87, i32 1, i32 31), !dbg !30
+  %89 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %86, i32 1, i32 31), !dbg !30
+  %90 = insertelement <2 x i32> poison, i32 %88, i64 0, !dbg !30
+  %91 = insertelement <2 x i32> %90, i32 %89, i64 1, !dbg !30
+  %92 = bitcast <2 x i32> %91 to i64, !dbg !30
+  %93 = add i64 %85, %92, !dbg !26
+  %94 = and i32 %11, 899, !dbg !30
+  %95 = icmp eq i32 %94, 0, !dbg !30
+  %96 = insertelement <1 x i64> poison, i64 %93, i64 0, !dbg !30
+  tail call void asm sideeffect "@$2 st.shared.b64 [ $0 + 0 ], $1;", "r,l,b"(ptr addrspace(3) %76, <1 x i64> %96, i1 %95) #5, !dbg !30
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !30
+  %97 = load i64, ptr addrspace(3) %72, align 16, !dbg !30
+  %98 = trunc i64 %97 to i32, !dbg !31
+  %99 = icmp slt i64 %2, 2, !dbg !32
+  %100 = icmp sgt i64 %2, 1, !dbg !33
+  %101 = select i1 %100, i64 %2, i64 0, !dbg !34
+  %102 = zext i1 %99 to i64, !dbg !35
+  %103 = add i64 %101, %102, !dbg !36
+  %104 = mul i64 %18, %103, !dbg !37
+  %105 = getelementptr i32, ptr addrspace(1) %1, i64 %.decomposed, !dbg !38
+  %106 = getelementptr i32, ptr addrspace(1) %105, i64 %104, !dbg !38
+  %107 = and i32 %11, 96, !dbg !39
+  %108 = icmp eq i32 %107, 0, !dbg !39
+  %109 = and i1 %108, %14, !dbg !39
+  tail call void asm sideeffect "@$2 st.global.b32 [ $1 + 0 ], { $0 };", "r,l,b"(i32 %98, ptr addrspace(1) %106, i1 %109) #5, !dbg !39
+  ret void, !dbg !40
+}
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 2147483647) i32 @llvm.nvvm.read.ptx.sreg.ctaid.x() #1
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 1024) i32 @llvm.nvvm.read.ptx.sreg.tid.x() #1
+
+; Function Attrs: convergent nocallback nounwind
+declare void @llvm.nvvm.barrier.cta.sync.aligned.all(i32) #2
+
+; Function Attrs: convergent nocallback nounwind memory(inaccessiblemem: readwrite)
+declare i32 @llvm.nvvm.shfl.sync.bfly.i32(i32, i32, i32, i32) #3
+
+; Function Attrs: nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare i64 @llvm.vector.reduce.add.v4i64(<4 x i64>) #4
+
+attributes #0 = { nounwind "nvvm.reqntid"="128" }
+attributes #1 = { mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #2 = { convergent nocallback nounwind }
+attributes #3 = { convergent nocallback nounwind memory(inaccessiblemem: readwrite) }
+attributes #4 = { nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #5 = { nounwind }
+
+!llvm.dbg.cu = !{!0}
+!llvm.module.flags = !{!2, !3}
+
+!0 = distinct !DICompileUnit(language: DW_LANG_C, file: !1, producer: "triton", isOptimized: true, runtimeVersion: 0, emissionKind: LineTablesOnly)
+!1 = !DIFile(filename: "cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py", directory: "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc")
+!2 = !{i32 2, !"Debug Info Version", i32 3}
+!3 = !{i32 4, !"nvvm-reflect-ftz", i32 1}
+!4 = distinct !DISubprogram(name: "triton_red_fused__to_copy_clone_slice_sum_transpose_5", linkageName: "triton_red_fused__to_copy_clone_slice_sum_transpose_5", scope: !1, file: !1, line: 18, type: !5, scopeLine: 18, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0)
+!5 = !DISubroutineType(cc: DW_CC_normal, types: !6)
+!6 = !{}
+!7 = !DILocation(line: 21, column: 28, scope: !4)
+!8 = !DILocation(line: 21, column: 33, scope: !4)
+!9 = !DILocation(line: 22, column: 44, scope: !4)
+!10 = !DILocation(line: 22, column: 23, scope: !4)
+!11 = !DILocation(line: 23, column: 21, scope: !4)
+!12 = !DILocation(line: 24, column: 37, scope: !4)
+!13 = !DILocation(line: 26, column: 19, scope: !4)
+!14 = !DILocation(line: 27, column: 19, scope: !4)
+!15 = !DILocation(line: 30, column: 40, scope: !4)
+!16 = !DILocation(line: 36, column: 54, scope: !4)
+!17 = !DILocation(line: 36, column: 58, scope: !4)
+!18 = !DILocation(line: 36, column: 73, scope: !4)
+!19 = !DILocation(line: 32, column: 29, scope: !4)
+!20 = !DILocation(line: 36, column: 63, scope: !4)
+!21 = !DILocation(line: 31, column: 31, scope: !4)
+!22 = !DILocation(line: 36, column: 43, scope: !4)
+!23 = !DILocation(line: 36, column: 34, scope: !4)
+!24 = !DILocation(line: 37, column: 23, scope: !4)
+!25 = !DILocation(line: 40, column: 48, scope: !4)
+!26 = !DILocation(line: 261, column: 15, scope: !27, inlinedAt: !29)
+!27 = distinct !DILexicalBlockFile(scope: !4, file: !28, discriminator: 0)
+!28 = !DIFile(filename: "standard.py", directory: "/workspace/specforge/lib/python3.11/site-packages/triton/language")
+!29 = !DILocation(line: 41, column: 25, scope: !4)
+!30 = !DILocation(line: 291, column: 36, scope: !27, inlinedAt: !29)
+!31 = !DILocation(line: 42, column: 19, scope: !4)
+!32 = !DILocation(line: 43, column: 49, scope: !4)
+!33 = !DILocation(line: 43, column: 75, scope: !4)
+!34 = !DILocation(line: 43, column: 66, scope: !4)
+!35 = !DILocation(line: 43, scope: !4)
+!36 = !DILocation(line: 43, column: 57, scope: !4)
+!37 = !DILocation(line: 43, column: 34, scope: !4)
+!38 = !DILocation(line: 43, column: 25, scope: !4)
+!39 = !DILocation(line: 43, column: 88, scope: !4)
+!40 = !DILocation(line: 43, column: 4, scope: !4)

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx

@@ -0,0 +1,525 @@
+//
+// Generated by LLVM NVPTX Back-End
+//
+
+.version 8.7
+.target sm_90a
+.address_size 64
+
+	// .globl	triton_red_fused__to_copy_clone_slice_sum_transpose_5 // -- Begin function triton_red_fused__to_copy_clone_slice_sum_transpose_5
+.extern .shared .align 16 .b8 global_smem[];
+                                        // @triton_red_fused__to_copy_clone_slice_sum_transpose_5
+.visible .entry triton_red_fused__to_copy_clone_slice_sum_transpose_5(
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_0,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_1,
+	.param .u64 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_2,
+	.param .u64 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_3,
+	.param .u32 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_4,
+	.param .u32 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_5,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_6,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_7
+)
+.reqntid 128
+{
+	.reg .pred 	%p<24>;
+	.reg .b32 	%r<51>;
+	.reg .b64 	%rd<97>;
+	.loc	1 18 0                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:18:0
+$L__func_begin0:
+	.loc	1 18 0                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:18:0
+
+// %bb.0:
+	ld.param.b32 	%r11, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_5];
+	ld.param.b64 	%rd20, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_2];
+$L__tmp0:
+	.loc	1 21 28                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:21:28
+	mov.u32 	%r12, %ctaid.x;
+	.loc	1 21 33                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:21:33
+	shl.b32 	%r13, %r12, 5;
+	.loc	1 22 44                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:22:44
+	mov.u32 	%r1, %tid.x;
+	and.b32 	%r2, %r1, 31;
+	.loc	1 22 23                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:22:23
+	or.b32 	%r14, %r13, %r2;
+	.loc	1 26 19                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:26:19
+	cvt.s64.s32 	%rd1, %r14;
+	.loc	1 27 19                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:27:19
+	or.b64 	%rd23, %rd1, %rd20;
+	and.b64 	%rd24, %rd23, -4294967296;
+	setp.ne.b64 	%p5, %rd24, 0;
+	cvt.u32.u64 	%r49, %rd1;
+	@%p5 bra 	$L__BB0_2;
+	bra.uni 	$L__BB0_1;
+$L__BB0_2:
+	div.s64 	%rd91, %rd1, %rd20;
+	bra.uni 	$L__BB0_3;
+$L__BB0_1:
+	cvt.u32.u64 	%r15, %rd20;
+	div.u32 	%r17, %r49, %r15;
+	cvt.u64.u32 	%rd91, %r17;
+$L__BB0_3:
+	.loc	1 0 19                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:0:19
+	ld.param.b32 	%r10, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_4];
+	ld.param.b64 	%rd19, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_1];
+	bfe.u32 	%r3, %r1, 5, 2;
+	.loc	1 26 19                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:26:19
+	mul.lo.s64 	%rd26, %rd91, %rd20;
+	sub.s64 	%rd6, %rd1, %rd26;
+	.loc	1 30 40                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:30:40
+	setp.lt.s32 	%p6, %r11, 1;
+	mov.b64 	%rd96, 0;
+	shl.b64 	%rd90, %rd6, 2;
+	@%p6 bra 	$L__BB0_7;
+// %bb.4:                               // %.lr.ph
+	.loc	1 0 40                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:0:40
+	ld.param.b64 	%rd21, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_3];
+	ld.param.b64 	%rd18, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_0];
+	.loc	1 23 21                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:23:21
+	setp.lt.s32 	%p1, %r49, %r10;
+	.loc	1 36 54                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:54
+	mul.lo.s64 	%rd31, %rd21, %rd20;
+	.loc	1 36 58                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:58
+	mul.lo.s64 	%rd32, %rd31, %rd91;
+	add.s64 	%rd34, %rd18, %rd90;
+	.loc	1 30 40                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:30:40
+	shl.b64 	%rd35, %rd32, 2;
+	add.s64 	%rd7, %rd34, %rd35;
+	mov.b64 	%rd92, 0;
+	mov.b32 	%r50, 0;
+	mov.b64 	%rd93, %rd92;
+	mov.b64 	%rd94, %rd92;
+	mov.b64 	%rd95, %rd92;
+$L__BB0_5:                              // =>This Inner Loop Header: Depth=1
+	.loc	1 36 63                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:63
+	// begin inline asm
+	mov.u64 %rd36, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd36, 1.0;
+	// end inline asm
+	.loc	1 31 31                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:31:31
+	add.s32 	%r24, %r3, %r50;
+	add.s32 	%r25, %r24, 4;
+	add.s32 	%r26, %r24, 8;
+	.loc	1 32 29                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:32:29
+	add.s32 	%r27, %r24, 12;
+	setp.lt.s32 	%p11, %r24, %r11;
+	setp.lt.s32 	%p12, %r25, %r11;
+	setp.lt.s32 	%p13, %r26, %r11;
+	setp.lt.s32 	%p14, %r27, %r11;
+	.loc	1 36 43                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:43
+	cvt.s64.s32 	%rd48, %r24;
+	cvt.s64.s32 	%rd49, %r25;
+	cvt.s64.s32 	%rd50, %r26;
+	cvt.s64.s32 	%rd51, %r27;
+	mul.lo.s64 	%rd52, %rd20, %rd48;
+	mul.lo.s64 	%rd53, %rd20, %rd49;
+	mul.lo.s64 	%rd54, %rd20, %rd50;
+	mul.lo.s64 	%rd55, %rd20, %rd51;
+	.loc	1 36 34                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:34
+	shl.b64 	%rd56, %rd52, 2;
+	add.s64 	%rd37, %rd7, %rd56;
+	shl.b64 	%rd57, %rd53, 2;
+	add.s64 	%rd40, %rd7, %rd57;
+	shl.b64 	%rd58, %rd54, 2;
+	add.s64 	%rd43, %rd7, %rd58;
+	shl.b64 	%rd59, %rd55, 2;
+	add.s64 	%rd46, %rd7, %rd59;
+	.loc	1 36 73                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:73
+	and.pred 	%p10, %p1, %p14;
+	and.pred 	%p9, %p1, %p13;
+	and.pred 	%p8, %p1, %p12;
+	and.pred 	%p7, %p1, %p11;
+	.loc	1 36 63                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:36:63
+	// begin inline asm
+	mov.u32 %r20, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b32 { %r20 }, [ %rd37 + 0 ], %rd36;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd39, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd39, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r21, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b32 { %r21 }, [ %rd40 + 0 ], %rd39;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd42, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd42, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r22, 0x0;
+	@%p9 ld.global.L1::evict_last.L2::cache_hint.b32 { %r22 }, [ %rd43 + 0 ], %rd42;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd45, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd45, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r23, 0x0;
+	@%p10 ld.global.L1::evict_last.L2::cache_hint.b32 { %r23 }, [ %rd46 + 0 ], %rd45;
+	// end inline asm
+	.loc	1 37 23                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:37:23
+	cvt.s64.s32 	%rd60, %r20;
+	cvt.s64.s32 	%rd61, %r21;
+	cvt.s64.s32 	%rd62, %r22;
+	cvt.s64.s32 	%rd63, %r23;
+	.loc	1 40 48                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:40:48
+	selp.b64 	%rd64, %rd63, 0, %p10;
+	selp.b64 	%rd65, %rd62, 0, %p9;
+	selp.b64 	%rd66, %rd61, 0, %p8;
+	selp.b64 	%rd67, %rd60, 0, %p7;
+	add.s64 	%rd92, %rd67, %rd92;
+	add.s64 	%rd93, %rd66, %rd93;
+	add.s64 	%rd94, %rd65, %rd94;
+	add.s64 	%rd95, %rd64, %rd95;
+	.loc	1 30 40                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:30:40
+	add.s32 	%r50, %r50, 16;
+	setp.lt.s32 	%p15, %r50, %r11;
+	@%p15 bra 	$L__BB0_5;
+// %bb.6:                               // %._crit_edge.loopexit
+$L__tmp1:
+	.loc	2 261 15                        // standard.py:261:15 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	add.s64 	%rd68, %rd92, %rd94;
+	add.s64 	%rd69, %rd93, %rd95;
+	add.s64 	%rd96, %rd68, %rd69;
+$L__tmp2:
+$L__BB0_7:                              // %._crit_edge
+	.loc	1 23 21                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:23:21
+	setp.lt.s32 	%p20, %r49, %r10;
+$L__tmp3:
+	.loc	2 291 36                        // standard.py:291:36 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	shl.b32 	%r33, %r2, 5;
+	mov.b32 	%r34, global_smem;
+	add.s32 	%r35, %r34, %r33;
+	shl.b32 	%r36, %r3, 3;
+	add.s32 	%r28, %r35, %r36;
+	mov.pred 	%p16, -1;
+	// begin inline asm
+	@%p16 st.shared.b64 [ %r28 + 0 ], %rd96;
+	// end inline asm
+	bar.sync 	0;
+	setp.lt.u32 	%p17, %r1, 128;
+	shl.b32 	%r37, %r1, 3;
+	add.s32 	%r29, %r34, %r37;
+	// begin inline asm
+	@%p17 ld.shared.b64 %rd71, [ %r29 + 0 ];
+	// end inline asm
+	mov.b64 	{_, %r38}, %rd71;
+	cvt.u32.u64 	%r39, %rd71;
+	shfl.sync.bfly.b32 	%r40, %r39, 2, 31, -1;
+	shfl.sync.bfly.b32 	%r41, %r38, 2, 31, -1;
+	cvt.u64.u32 	%rd74, %r40;
+	cvt.u64.u32 	%rd75, %r41;
+	shl.b64 	%rd76, %rd75, 32;
+	or.b64 	%rd77, %rd74, %rd76;
+	.loc	2 261 15                        // standard.py:261:15 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	add.s64 	%rd78, %rd71, %rd77;
+	.loc	2 291 36                        // standard.py:291:36 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	mov.b64 	{_, %r42}, %rd78;
+	cvt.u32.u64 	%r43, %rd78;
+	shfl.sync.bfly.b32 	%r44, %r43, 1, 31, -1;
+	shfl.sync.bfly.b32 	%r45, %r42, 1, 31, -1;
+	cvt.u64.u32 	%rd79, %r44;
+	cvt.u64.u32 	%rd80, %r45;
+	shl.b64 	%rd81, %rd80, 32;
+	or.b64 	%rd82, %rd79, %rd81;
+	.loc	2 261 15                        // standard.py:261:15 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	add.s64 	%rd72, %rd78, %rd82;
+	.loc	2 291 36                        // standard.py:291:36 @[ cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:41:25 ]
+	and.b32 	%r46, %r1, 899;
+	setp.eq.b32 	%p18, %r46, 0;
+	// begin inline asm
+	@%p18 st.shared.b64 [ %r29 + 0 ], %rd72;
+	// end inline asm
+	bar.sync 	0;
+	ld.shared.b32 	%r31, [%r35];
+$L__tmp4:
+	.loc	1 43 49                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:49
+	setp.lt.s64 	%p21, %rd20, 2;
+	.loc	1 43 75                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:75
+	setp.gt.s64 	%p22, %rd20, 1;
+	.loc	1 43 66                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:66
+	selp.b64 	%rd83, %rd20, 0, %p22;
+	.loc	1 43 0                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43
+	selp.b64 	%rd84, 1, 0, %p21;
+	.loc	1 43 57                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:57
+	add.s64 	%rd85, %rd83, %rd84;
+	.loc	1 43 34                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:34
+	mul.lo.s64 	%rd86, %rd91, %rd85;
+	.loc	1 43 25                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:25
+	add.s64 	%rd88, %rd19, %rd90;
+	shl.b64 	%rd89, %rd86, 2;
+	add.s64 	%rd73, %rd88, %rd89;
+	.loc	1 43 88                         // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:88
+	and.b32 	%r47, %r1, 96;
+	setp.eq.b32 	%p23, %r47, 0;
+	and.pred 	%p19, %p23, %p20;
+	// begin inline asm
+	@%p19 st.global.b32 [ %rd73 + 0 ], { %r31 };
+	// end inline asm
+	.loc	1 43 4                          // cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py:43:4
+	ret;
+$L__tmp5:
+$L__func_end0:
+                                        // -- End function
+}
+	.file	1 "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py"
+	.file	2 "/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py"
+	.section	.debug_abbrev
+	{
+.b8 1                                   // Abbreviation Code
+.b8 17                                  // DW_TAG_compile_unit
+.b8 1                                   // DW_CHILDREN_yes
+.b8 37                                  // DW_AT_producer
+.b8 8                                   // DW_FORM_string
+.b8 19                                  // DW_AT_language
+.b8 5                                   // DW_FORM_data2
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 16                                  // DW_AT_stmt_list
+.b8 6                                   // DW_FORM_data4
+.b8 27                                  // DW_AT_comp_dir
+.b8 8                                   // DW_FORM_string
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 2                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 0                                   // DW_CHILDREN_no
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 32                                  // DW_AT_inline
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 3                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 1                                   // DW_CHILDREN_yes
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 4                                   // Abbreviation Code
+.b8 29                                  // DW_TAG_inlined_subroutine
+.b8 0                                   // DW_CHILDREN_no
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 88                                  // DW_AT_call_file
+.b8 11                                  // DW_FORM_data1
+.b8 89                                  // DW_AT_call_line
+.b8 11                                  // DW_FORM_data1
+.b8 87                                  // DW_AT_call_column
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 0                                   // EOM(3)
+	}
+	.section	.debug_info
+	{
+.b32 238                                // Length of Unit
+.b8 2                                   // DWARF version number
+.b8 0
+.b32 .debug_abbrev                      // Offset Into Abbrev. Section
+.b8 8                                   // Address Size (in bytes)
+.b8 1                                   // Abbrev [1] 0xb:0xe7 DW_TAG_compile_unit
+.b8 116                                 // DW_AT_producer
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 0
+.b8 2                                   // DW_AT_language
+.b8 0
+.b8 99                                  // DW_AT_name
+.b8 119
+.b8 99
+.b8 50
+.b8 106
+.b8 99
+.b8 116
+.b8 122
+.b8 54
+.b8 55
+.b8 115
+.b8 55
+.b8 110
+.b8 54
+.b8 102
+.b8 111
+.b8 116
+.b8 108
+.b8 50
+.b8 50
+.b8 103
+.b8 114
+.b8 52
+.b8 52
+.b8 103
+.b8 108
+.b8 105
+.b8 106
+.b8 114
+.b8 99
+.b8 102
+.b8 120
+.b8 112
+.b8 101
+.b8 98
+.b8 108
+.b8 118
+.b8 107
+.b8 51
+.b8 50
+.b8 104
+.b8 111
+.b8 114
+.b8 110
+.b8 108
+.b8 108
+.b8 108
+.b8 122
+.b8 110
+.b8 120
+.b8 50
+.b8 109
+.b8 46
+.b8 112
+.b8 121
+.b8 0
+.b32 .debug_line                        // DW_AT_stmt_list
+.b8 47                                  // DW_AT_comp_dir
+.b8 119
+.b8 111
+.b8 114
+.b8 107
+.b8 115
+.b8 112
+.b8 97
+.b8 99
+.b8 101
+.b8 47
+.b8 104
+.b8 97
+.b8 110
+.b8 114
+.b8 117
+.b8 105
+.b8 47
+.b8 83
+.b8 112
+.b8 101
+.b8 99
+.b8 70
+.b8 111
+.b8 114
+.b8 103
+.b8 101
+.b8 45
+.b8 101
+.b8 120
+.b8 116
+.b8 47
+.b8 99
+.b8 97
+.b8 99
+.b8 104
+.b8 101
+.b8 47
+.b8 99
+.b8 111
+.b8 109
+.b8 112
+.b8 105
+.b8 108
+.b8 101
+.b8 100
+.b8 95
+.b8 107
+.b8 101
+.b8 114
+.b8 110
+.b8 101
+.b8 108
+.b8 115
+.b8 47
+.b8 119
+.b8 99
+.b8 0
+.b8 2                                   // Abbrev [2] 0x8b:0x38 DW_TAG_subprogram
+.b8 116                                 // DW_AT_name
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 95
+.b8 114
+.b8 101
+.b8 100
+.b8 95
+.b8 102
+.b8 117
+.b8 115
+.b8 101
+.b8 100
+.b8 95
+.b8 95
+.b8 116
+.b8 111
+.b8 95
+.b8 99
+.b8 111
+.b8 112
+.b8 121
+.b8 95
+.b8 99
+.b8 108
+.b8 111
+.b8 110
+.b8 101
+.b8 95
+.b8 115
+.b8 108
+.b8 105
+.b8 99
+.b8 101
+.b8 95
+.b8 115
+.b8 117
+.b8 109
+.b8 95
+.b8 116
+.b8 114
+.b8 97
+.b8 110
+.b8 115
+.b8 112
+.b8 111
+.b8 115
+.b8 101
+.b8 95
+.b8 53
+.b8 0
+.b8 1                                   // DW_AT_inline
+.b8 3                                   // Abbrev [3] 0xc3:0x2e DW_TAG_subprogram
+.b64 $L__func_begin0                    // DW_AT_low_pc
+.b64 $L__func_end0                      // DW_AT_high_pc
+.b32 139                                // DW_AT_abstract_origin
+.b8 4                                   // Abbrev [4] 0xd8:0x18 DW_TAG_inlined_subroutine
+.b32 139                                // DW_AT_abstract_origin
+.b64 $L__tmp1                           // DW_AT_low_pc
+.b64 $L__tmp4                           // DW_AT_high_pc
+.b8 1                                   // DW_AT_call_file
+.b8 41                                  // DW_AT_call_line
+.b8 25                                  // DW_AT_call_column
+.b8 0                                   // End Of Children Mark
+.b8 0                                   // End Of Children Mark
+	}
+	.section	.debug_macinfo	{	}

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.source

@@ -0,0 +1,193 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":18:0)
+#loc41 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":285:0)
+#loc43 = loc(unknown)
+#loc46 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":260:0)
+#loc50 = loc("in_ptr0"(#loc))
+#loc51 = loc("out_ptr1"(#loc))
+#loc52 = loc("ks0"(#loc))
+#loc53 = loc("ks1"(#loc))
+#loc54 = loc("xnumel"(#loc))
+#loc55 = loc("r0_numel"(#loc))
+#loc85 = loc("input"(#loc41))
+#loc86 = loc("a"(#loc46))
+#loc87 = loc("b"(#loc46))
+module {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %xoffset = tt.get_program_id x : i32 loc(#loc56)
+    %xoffset_0 = arith.constant 32 : i32 loc(#loc57)
+    %xoffset_1 = arith.constant 32 : i32 loc(#loc57)
+    %xoffset_2 = arith.muli %xoffset, %xoffset_1 : i32 loc(#loc57)
+    %xindex = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32> loc(#loc58)
+    %xindex_3 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<32xi32> -> tensor<32x1xi32> loc(#loc59)
+    %xindex_4 = tt.splat %xoffset_2 : i32 -> tensor<32x1xi32> loc(#loc60)
+    %xindex_5 = arith.addi %xindex_4, %xindex_3 : tensor<32x1xi32> loc(#loc60)
+    %xmask = tt.splat %xnumel : i32 -> tensor<32x1xi32> loc(#loc61)
+    %xmask_6 = arith.cmpi slt, %xindex_5, %xmask : tensor<32x1xi32> loc(#loc61)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32> loc(#loc62)
+    %r0_base_7 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32> -> tensor<1x16xi32> loc(#loc63)
+    %x0 = arith.extsi %xindex_5 : tensor<32x1xi32> to tensor<32x1xi64> loc(#loc64)
+    %x0_8 = tt.splat %ks0 : i64 -> tensor<32x1xi64> loc(#loc64)
+    %x0_9 = arith.remsi %x0, %x0_8 : tensor<32x1xi64> loc(#loc64)
+    %x1 = arith.extsi %xindex_5 : tensor<32x1xi32> to tensor<32x1xi64> loc(#loc65)
+    %x1_10 = tt.splat %ks0 : i64 -> tensor<32x1xi64> loc(#loc65)
+    %x1_11 = arith.divsi %x1, %x1_10 : tensor<32x1xi64> loc(#loc65)
+    %_tmp3 = arith.constant 0 : i64 loc(#loc66)
+    %_tmp3_12 = arith.constant dense<0> : tensor<32x16xi64> loc(#loc66)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc12)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc12)
+    %0 = arith.bitcast %c0_i32 : i32 to i32 loc(#loc12)
+    %1 = arith.bitcast %r0_numel : i32 to i32 loc(#loc12)
+    %2 = arith.bitcast %c16_i32 : i32 to i32 loc(#loc12)
+    %3 = ub.poison : i32 loc(#loc12)
+    %_tmp3_13 = scf.for %r0_offset = %0 to %1 step %2 iter_args(%_tmp3_18 = %_tmp3_12) -> (tensor<32x16xi64>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x16xi32> loc(#loc68)
+      %r0_index_19 = arith.addi %r0_index, %r0_base_7 : tensor<1x16xi32> loc(#loc68)
+      %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32> loc(#loc69)
+      %r0_mask_20 = arith.cmpi slt, %r0_index_19, %r0_mask : tensor<1x16xi32> loc(#loc69)
+      %tmp0 = arith.extsi %r0_index_19 : tensor<1x16xi32> to tensor<1x16xi64> loc(#loc70)
+      %tmp0_21 = tt.splat %ks0 : i64 -> tensor<1x16xi64> loc(#loc70)
+      %tmp0_22 = arith.muli %tmp0_21, %tmp0 : tensor<1x16xi64> loc(#loc70)
+      %tmp0_23 = tt.broadcast %x0_9 : tensor<32x1xi64> -> tensor<32x16xi64> loc(#loc71)
+      %tmp0_24 = tt.broadcast %tmp0_22 : tensor<1x16xi64> -> tensor<32x16xi64> loc(#loc71)
+      %tmp0_25 = arith.addi %tmp0_23, %tmp0_24 : tensor<32x16xi64> loc(#loc71)
+      %tmp0_26 = arith.muli %ks0, %ks1 : i64 loc(#loc72)
+      %tmp0_27 = tt.splat %tmp0_26 : i64 -> tensor<32x1xi64> loc(#loc73)
+      %tmp0_28 = arith.muli %tmp0_27, %x1_11 : tensor<32x1xi64> loc(#loc73)
+      %tmp0_29 = tt.broadcast %tmp0_28 : tensor<32x1xi64> -> tensor<32x16xi64> loc(#loc74)
+      %tmp0_30 = arith.addi %tmp0_25, %tmp0_29 : tensor<32x16xi64> loc(#loc74)
+      %tmp0_31 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>> loc(#loc75)
+      %tmp0_32 = tt.addptr %tmp0_31, %tmp0_30 : tensor<32x16x!tt.ptr<i32>>, tensor<32x16xi64> loc(#loc75)
+      %tmp0_33 = tt.broadcast %r0_mask_20 : tensor<1x16xi1> -> tensor<32x16xi1> loc(#loc76)
+      %tmp0_34 = tt.broadcast %xmask_6 : tensor<32x1xi1> -> tensor<32x16xi1> loc(#loc76)
+      %tmp0_35 = arith.andi %tmp0_33, %tmp0_34 : tensor<32x16xi1> loc(#loc76)
+      %tmp0_36 = arith.constant 0.000000e+00 : f32 loc(#loc77)
+      %tmp0_37 = arith.constant dense<0.000000e+00> : tensor<32x16xf32> loc(#loc77)
+      %tmp0_38 = arith.fptosi %tmp0_37 : tensor<32x16xf32> to tensor<32x16xi32> loc(#loc77)
+      %tmp0_39 = tt.load %tmp0_32, %tmp0_35, %tmp0_38 evictionPolicy = evict_last : tensor<32x16x!tt.ptr<i32>> loc(#loc77)
+      %tmp1 = arith.extsi %tmp0_39 : tensor<32x16xi32> to tensor<32x16xi64> loc(#loc78)
+      %tmp4 = arith.addi %_tmp3_18, %tmp1 : tensor<32x16xi64> loc(#loc79)
+      %_tmp3_40 = tt.broadcast %r0_mask_20 : tensor<1x16xi1> -> tensor<32x16xi1> loc(#loc80)
+      %_tmp3_41 = tt.broadcast %xmask_6 : tensor<32x1xi1> -> tensor<32x16xi1> loc(#loc80)
+      %_tmp3_42 = arith.andi %_tmp3_40, %_tmp3_41 : tensor<32x16xi1> loc(#loc80)
+      %_tmp3_43 = arith.select %_tmp3_42, %tmp4, %_tmp3_18 : tensor<32x16xi1>, tensor<32x16xi64> loc(#loc81)
+      scf.yield %_tmp3_43 : tensor<32x16xi64> loc(#loc27)
+    } loc(#loc67)
+    %tmp3 = tt.call @"triton.language.standard.sum__i64S32_16S__(1,)cconstexpr_1__(2,)cconstexpr_False__(3,)cNone"(%_tmp3_13) : (tensor<32x16xi64>) -> tensor<32xi64> loc(#loc82)
+    %tmp3_14 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<32xi64> -> tensor<32x1xi64> loc(#loc83)
+    %tmp5 = arith.trunci %tmp3_14 : tensor<32x1xi64> to tensor<32x1xi32> loc(#loc84)
+    %c1_i32 = arith.constant 1 : i32 loc(#loc31)
+    %4 = arith.extsi %c1_i32 : i32 to i64 loc(#loc31)
+    %5 = arith.cmpi sge, %4, %ks0 : i64 loc(#loc31)
+    %c1_i32_15 = arith.constant 1 : i32 loc(#loc32)
+    %c1_i32_16 = arith.constant 1 : i32 loc(#loc32)
+    %6 = arith.extui %5 : i1 to i32 loc(#loc32)
+    %7 = arith.muli %c1_i32_16, %6 : i32 loc(#loc32)
+    %c1_i32_17 = arith.constant 1 : i32 loc(#loc33)
+    %8 = arith.extsi %c1_i32_17 : i32 to i64 loc(#loc33)
+    %9 = arith.cmpi sgt, %ks0, %8 : i64 loc(#loc33)
+    %10 = arith.extui %9 : i1 to i64 loc(#loc34)
+    %11 = arith.muli %ks0, %10 : i64 loc(#loc34)
+    %12 = arith.extsi %7 : i32 to i64 loc(#loc35)
+    %13 = arith.addi %12, %11 : i64 loc(#loc35)
+    %14 = tt.splat %13 : i64 -> tensor<32x1xi64> loc(#loc36)
+    %15 = arith.muli %x1_11, %14 : tensor<32x1xi64> loc(#loc36)
+    %16 = arith.addi %x0_9, %15 : tensor<32x1xi64> loc(#loc37)
+    %17 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<32x1x!tt.ptr<i32>> loc(#loc38)
+    %18 = tt.addptr %17, %16 : tensor<32x1x!tt.ptr<i32>>, tensor<32x1xi64> loc(#loc38)
+    tt.store %18, %tmp5, %xmask_6 : tensor<32x1x!tt.ptr<i32>> loc(#loc39)
+    tt.return loc(#loc40)
+  } loc(#loc)
+  tt.func private @"triton.language.standard.sum__i64S32_16S__(1,)cconstexpr_1__(2,)cconstexpr_False__(3,)cNone"(%input: tensor<32x16xi64> loc("input"(#loc41))) -> tensor<32xi64> attributes {noinline = false} {
+    %0 = "tt.reduce"(%input) <{axis = 1 : i32}> ({
+    ^bb0(%arg1: i64 loc(unknown), %arg2: i64 loc(unknown)):
+      %2 = tt.call @triton.language.standard._sum_combine__i64_i64__(%arg1, %arg2) : (i64, i64) -> i64 loc(#loc42)
+      tt.reduce.return %2 : i64 loc(#loc42)
+    }) : (tensor<32x16xi64>) -> tensor<32xi64> loc(#loc42)
+    tt.return %0 : tensor<32xi64> loc(#loc44)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : tensor<32xi64> loc(#loc45)
+    tt.return %1 : tensor<32xi64> loc(#loc45)
+  } loc(#loc41)
+  tt.func private @triton.language.standard._sum_combine__i64_i64__(%a: i64 loc("a"(#loc46)), %b: i64 loc("b"(#loc46))) -> i64 attributes {noinline = false} {
+    %0 = arith.addi %a, %b : i64 loc(#loc47)
+    tt.return %0 : i64 loc(#loc48)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : i64 loc(#loc49)
+    tt.return %1 : i64 loc(#loc49)
+  } loc(#loc46)
+} loc(#loc)
+#loc1 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:28)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:33)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:36)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":23:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":24:27)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":24:37)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":26:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":27:19)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":28:43)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":30:40)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":31:31)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":32:29)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:43)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:39)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:54)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:58)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:50)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:34)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:73)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:63)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":37:23)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":39:23)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:35)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:48)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:8)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:25)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:28)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":42:19)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:49)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:41)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:75)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:66)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:57)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:34)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:30)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:25)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:88)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:4)
+#loc42 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc44 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:11)
+#loc45 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:4)
+#loc47 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc48 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:11)
+#loc49 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:4)
+#loc56 = loc("xoffset"(#loc1))
+#loc57 = loc("xoffset"(#loc2))
+#loc58 = loc("xindex"(#loc3))
+#loc59 = loc("xindex"(#loc4))
+#loc60 = loc("xindex"(#loc5))
+#loc61 = loc("xmask"(#loc6))
+#loc62 = loc("r0_base"(#loc7))
+#loc63 = loc("r0_base"(#loc8))
+#loc64 = loc("x0"(#loc9))
+#loc65 = loc("x1"(#loc10))
+#loc66 = loc("_tmp3"(#loc11))
+#loc67 = loc("_tmp3"(#loc12))
+#loc68 = loc("r0_index"(#loc13))
+#loc69 = loc("r0_mask"(#loc14))
+#loc70 = loc("tmp0"(#loc15))
+#loc71 = loc("tmp0"(#loc16))
+#loc72 = loc("tmp0"(#loc17))
+#loc73 = loc("tmp0"(#loc18))
+#loc74 = loc("tmp0"(#loc19))
+#loc75 = loc("tmp0"(#loc20))
+#loc76 = loc("tmp0"(#loc21))
+#loc77 = loc("tmp0"(#loc22))
+#loc78 = loc("tmp1"(#loc23))
+#loc79 = loc("tmp4"(#loc24))
+#loc80 = loc("_tmp3"(#loc25))
+#loc81 = loc("_tmp3"(#loc26))
+#loc82 = loc("tmp3"(#loc28))
+#loc83 = loc("tmp3"(#loc29))
+#loc84 = loc("tmp5"(#loc30))

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir

@@ -0,0 +1,147 @@
+#blocked = #ttg.blocked<{sizePerThread = [1, 1], threadsPerWarp = [32, 1], warpsPerCTA = [1, 4], order = [0, 1]}>
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":18:0)
+#loc1 = loc(unknown)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:25)
+#loc40 = loc("in_ptr0"(#loc))
+#loc41 = loc("out_ptr1"(#loc))
+#loc42 = loc("ks0"(#loc))
+#loc43 = loc("ks1"(#loc))
+#loc44 = loc("xnumel"(#loc))
+#loc45 = loc("r0_numel"(#loc))
+#loc68 = loc("tmp3"(#loc26))
+#loc73 = loc(callsite(#loc1 at #loc68))
+module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 4 : i32, ttg.target = "cuda:90", "ttg.threads-per-warp" = 32 : i32} {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<0> : tensor<32x16xi64, #blocked> loc(#loc1)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc1)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc1)
+    %c1_i64 = arith.constant 1 : i64 loc(#loc1)
+    %c32_i32 = arith.constant 32 : i32 loc(#loc1)
+    %cst_0 = arith.constant dense<0> : tensor<32x16xi32, #blocked> loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc46)
+    %xoffset_1 = arith.muli %xoffset, %c32_i32 : i32 loc(#loc47)
+    %xindex = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc48)
+    %xindex_2 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<32x1xi32, #blocked> loc(#loc48)
+    %xindex_3 = tt.splat %xoffset_1 : i32 -> tensor<32x1xi32, #blocked> loc(#loc49)
+    %xindex_4 = arith.addi %xindex_3, %xindex_2 : tensor<32x1xi32, #blocked> loc(#loc49)
+    %xmask = tt.splat %xnumel : i32 -> tensor<32x1xi32, #blocked> loc(#loc50)
+    %xmask_5 = arith.cmpi slt, %xindex_4, %xmask : tensor<32x1xi32, #blocked> loc(#loc50)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> loc(#loc51)
+    %r0_base_6 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> -> tensor<1x16xi32, #blocked> loc(#loc51)
+    %x0 = arith.extsi %xindex_4 : tensor<32x1xi32, #blocked> to tensor<32x1xi64, #blocked> loc(#loc52)
+    %x0_7 = tt.splat %ks0 : i64 -> tensor<32x1xi64, #blocked> loc(#loc52)
+    %x0_8 = arith.remsi %x0, %x0_7 : tensor<32x1xi64, #blocked> loc(#loc52)
+    %x1 = arith.divsi %x0, %x0_7 : tensor<32x1xi64, #blocked> loc(#loc53)
+    %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32, #blocked> loc(#loc54)
+    %tmp0 = tt.splat %ks0 : i64 -> tensor<1x16xi64, #blocked> loc(#loc55)
+    %tmp0_9 = tt.broadcast %x0_8 : tensor<32x1xi64, #blocked> -> tensor<32x16xi64, #blocked> loc(#loc56)
+    %tmp0_10 = arith.muli %ks0, %ks1 : i64 loc(#loc57)
+    %tmp0_11 = tt.splat %tmp0_10 : i64 -> tensor<32x1xi64, #blocked> loc(#loc58)
+    %tmp0_12 = arith.muli %tmp0_11, %x1 : tensor<32x1xi64, #blocked> loc(#loc58)
+    %tmp0_13 = tt.broadcast %tmp0_12 : tensor<32x1xi64, #blocked> -> tensor<32x16xi64, #blocked> loc(#loc59)
+    %tmp0_14 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>, #blocked> loc(#loc60)
+    %tmp0_15 = tt.broadcast %xmask_5 : tensor<32x1xi1, #blocked> -> tensor<32x16xi1, #blocked> loc(#loc61)
+    %_tmp3 = scf.for %_tmp3_17 = %c0_i32 to %r0_numel step %c16_i32 iter_args(%_tmp3_18 = %cst) -> (tensor<32x16xi64, #blocked>)  : i32 {
+      %r0_index = tt.splat %_tmp3_17 : i32 -> tensor<1x16xi32, #blocked> loc(#loc63)
+      %r0_index_19 = arith.addi %r0_index, %r0_base_6 : tensor<1x16xi32, #blocked> loc(#loc63)
+      %r0_mask_20 = arith.cmpi slt, %r0_index_19, %r0_mask : tensor<1x16xi32, #blocked> loc(#loc54)
+      %tmp0_21 = arith.extsi %r0_index_19 : tensor<1x16xi32, #blocked> to tensor<1x16xi64, #blocked> loc(#loc55)
+      %tmp0_22 = arith.muli %tmp0, %tmp0_21 : tensor<1x16xi64, #blocked> loc(#loc55)
+      %tmp0_23 = tt.broadcast %tmp0_22 : tensor<1x16xi64, #blocked> -> tensor<32x16xi64, #blocked> loc(#loc56)
+      %tmp0_24 = arith.addi %tmp0_9, %tmp0_23 : tensor<32x16xi64, #blocked> loc(#loc56)
+      %tmp0_25 = arith.addi %tmp0_24, %tmp0_13 : tensor<32x16xi64, #blocked> loc(#loc59)
+      %tmp0_26 = tt.addptr %tmp0_14, %tmp0_25 : tensor<32x16x!tt.ptr<i32>, #blocked>, tensor<32x16xi64, #blocked> loc(#loc60)
+      %tmp0_27 = tt.broadcast %r0_mask_20 : tensor<1x16xi1, #blocked> -> tensor<32x16xi1, #blocked> loc(#loc61)
+      %tmp0_28 = arith.andi %tmp0_27, %tmp0_15 : tensor<32x16xi1, #blocked> loc(#loc61)
+      %tmp0_29 = tt.load %tmp0_26, %tmp0_28, %cst_0 evictionPolicy = evict_last : tensor<32x16x!tt.ptr<i32>, #blocked> loc(#loc64)
+      %tmp1 = arith.extsi %tmp0_29 : tensor<32x16xi32, #blocked> to tensor<32x16xi64, #blocked> loc(#loc65)
+      %tmp4 = arith.addi %_tmp3_18, %tmp1 : tensor<32x16xi64, #blocked> loc(#loc66)
+      %_tmp3_30 = arith.select %tmp0_28, %tmp4, %_tmp3_18 : tensor<32x16xi1, #blocked>, tensor<32x16xi64, #blocked> loc(#loc67)
+      scf.yield %_tmp3_30 : tensor<32x16xi64, #blocked> loc(#loc24)
+    } loc(#loc62)
+    %tmp3 = "tt.reduce"(%_tmp3) <{axis = 1 : i32}> ({
+    ^bb0(%tmp3_17: i64 loc(callsite(#loc1 at #loc68)), %tmp3_18: i64 loc(callsite(#loc1 at #loc68))):
+      %tmp3_19 = arith.addi %tmp3_17, %tmp3_18 : i64 loc(#loc74)
+      tt.reduce.return %tmp3_19 : i64 loc(#loc72)
+    }) : (tensor<32x16xi64, #blocked>) -> tensor<32xi64, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc72)
+    %tmp3_16 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<32xi64, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<32x1xi64, #blocked> loc(#loc69)
+    %tmp5 = arith.trunci %tmp3_16 : tensor<32x1xi64, #blocked> to tensor<32x1xi32, #blocked> loc(#loc70)
+    %0 = arith.cmpi sle, %ks0, %c1_i64 : i64 loc(#loc30)
+    %1 = arith.cmpi sgt, %ks0, %c1_i64 : i64 loc(#loc31)
+    %2 = arith.extui %1 : i1 to i64 loc(#loc32)
+    %3 = arith.muli %ks0, %2 : i64 loc(#loc32)
+    %4 = arith.extui %0 : i1 to i64 loc(#loc71)
+    %5 = arith.addi %4, %3 : i64 loc(#loc33)
+    %6 = tt.splat %5 : i64 -> tensor<32x1xi64, #blocked> loc(#loc35)
+    %7 = arith.muli %x1, %6 : tensor<32x1xi64, #blocked> loc(#loc35)
+    %8 = arith.addi %x0_8, %7 : tensor<32x1xi64, #blocked> loc(#loc36)
+    %9 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<32x1x!tt.ptr<i32>, #blocked> loc(#loc37)
+    %10 = tt.addptr %9, %8 : tensor<32x1x!tt.ptr<i32>, #blocked>, tensor<32x1xi64, #blocked> loc(#loc37)
+    tt.store %10, %tmp5, %xmask_5 : tensor<32x1x!tt.ptr<i32>, #blocked> loc(#loc38)
+    tt.return loc(#loc39)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":23:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":24:37)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":26:19)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":27:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":32:29)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:43)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:39)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:54)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:58)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:50)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:34)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:73)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":30:40)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":31:31)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:63)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":37:23)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":39:23)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:48)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:8)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:28)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":42:19)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:49)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:75)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:66)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:57)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:41)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:34)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:30)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:25)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:88)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:4)
+#loc46 = loc("xoffset"(#loc2))
+#loc47 = loc("xoffset"(#loc3))
+#loc48 = loc("xindex"(#loc4))
+#loc49 = loc("xindex"(#loc5))
+#loc50 = loc("xmask"(#loc6))
+#loc51 = loc("r0_base"(#loc7))
+#loc52 = loc("x0"(#loc8))
+#loc53 = loc("x1"(#loc9))
+#loc54 = loc("r0_mask"(#loc10))
+#loc55 = loc("tmp0"(#loc11))
+#loc56 = loc("tmp0"(#loc12))
+#loc57 = loc("tmp0"(#loc13))
+#loc58 = loc("tmp0"(#loc14))
+#loc59 = loc("tmp0"(#loc15))
+#loc60 = loc("tmp0"(#loc16))
+#loc61 = loc("tmp0"(#loc17))
+#loc62 = loc("_tmp3"(#loc18))
+#loc63 = loc("r0_index"(#loc19))
+#loc64 = loc("tmp0"(#loc20))
+#loc65 = loc("tmp1"(#loc21))
+#loc66 = loc("tmp4"(#loc22))
+#loc67 = loc("_tmp3"(#loc23))
+#loc69 = loc("tmp3"(#loc28))
+#loc70 = loc("tmp5"(#loc29))
+#loc71 = loc(fused[#loc33, #loc34])
+#loc72 = loc(callsite(#loc25 at #loc68))
+#loc74 = loc(callsite(#loc27 at #loc72))

SpecForge-ext/cache/compiled_kernels/triton/0/2HBOMUT44J5WFCUWYGRFAAS3HGVNDHLHT7HCSXUCAOIKU6XGJNTA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir

@@ -0,0 +1,152 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":18:0)
+#loc1 = loc(unknown)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:25)
+#loc43 = loc("in_ptr0"(#loc))
+#loc44 = loc("out_ptr1"(#loc))
+#loc45 = loc("ks0"(#loc))
+#loc46 = loc("ks1"(#loc))
+#loc47 = loc("xnumel"(#loc))
+#loc48 = loc("r0_numel"(#loc))
+#loc74 = loc("tmp3"(#loc29))
+#loc79 = loc(callsite(#loc1 at #loc74))
+module {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %c1_i64 = arith.constant 1 : i64 loc(#loc1)
+    %cst = arith.constant dense<0> : tensor<32x16xi32> loc(#loc1)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc2)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc2)
+    %_tmp3 = arith.constant dense<0> : tensor<32x16xi64> loc(#loc49)
+    %c32_i32 = arith.constant 32 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc50)
+    %xoffset_0 = arith.muli %xoffset, %c32_i32 : i32 loc(#loc51)
+    %xindex = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32> loc(#loc52)
+    %xindex_1 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<32xi32> -> tensor<32x1xi32> loc(#loc53)
+    %xindex_2 = tt.splat %xoffset_0 : i32 -> tensor<32x1xi32> loc(#loc54)
+    %xindex_3 = arith.addi %xindex_2, %xindex_1 : tensor<32x1xi32> loc(#loc54)
+    %xmask = tt.splat %xnumel : i32 -> tensor<32x1xi32> loc(#loc55)
+    %xmask_4 = arith.cmpi slt, %xindex_3, %xmask : tensor<32x1xi32> loc(#loc55)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32> loc(#loc56)
+    %r0_base_5 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32> -> tensor<1x16xi32> loc(#loc57)
+    %x0 = arith.extsi %xindex_3 : tensor<32x1xi32> to tensor<32x1xi64> loc(#loc58)
+    %x0_6 = tt.splat %ks0 : i64 -> tensor<32x1xi64> loc(#loc58)
+    %x0_7 = arith.remsi %x0, %x0_6 : tensor<32x1xi64> loc(#loc58)
+    %x1 = arith.divsi %x0, %x0_6 : tensor<32x1xi64> loc(#loc59)
+    %_tmp3_8 = scf.for %r0_offset = %c0_i32 to %r0_numel step %c16_i32 iter_args(%_tmp3_10 = %_tmp3) -> (tensor<32x16xi64>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x16xi32> loc(#loc61)
+      %r0_index_11 = arith.addi %r0_index, %r0_base_5 : tensor<1x16xi32> loc(#loc61)
+      %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32> loc(#loc62)
+      %r0_mask_12 = arith.cmpi slt, %r0_index_11, %r0_mask : tensor<1x16xi32> loc(#loc62)
+      %tmp0 = arith.extsi %r0_index_11 : tensor<1x16xi32> to tensor<1x16xi64> loc(#loc63)
+      %tmp0_13 = tt.splat %ks0 : i64 -> tensor<1x16xi64> loc(#loc63)
+      %tmp0_14 = arith.muli %tmp0_13, %tmp0 : tensor<1x16xi64> loc(#loc63)
+      %tmp0_15 = tt.broadcast %x0_7 : tensor<32x1xi64> -> tensor<32x16xi64> loc(#loc64)
+      %tmp0_16 = tt.broadcast %tmp0_14 : tensor<1x16xi64> -> tensor<32x16xi64> loc(#loc64)
+      %tmp0_17 = arith.addi %tmp0_15, %tmp0_16 : tensor<32x16xi64> loc(#loc64)
+      %tmp0_18 = arith.muli %ks0, %ks1 : i64 loc(#loc65)
+      %tmp0_19 = tt.splat %tmp0_18 : i64 -> tensor<32x1xi64> loc(#loc66)
+      %tmp0_20 = arith.muli %tmp0_19, %x1 : tensor<32x1xi64> loc(#loc66)
+      %tmp0_21 = tt.broadcast %tmp0_20 : tensor<32x1xi64> -> tensor<32x16xi64> loc(#loc67)
+      %tmp0_22 = arith.addi %tmp0_17, %tmp0_21 : tensor<32x16xi64> loc(#loc67)
+      %tmp0_23 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>> loc(#loc68)
+      %tmp0_24 = tt.addptr %tmp0_23, %tmp0_22 : tensor<32x16x!tt.ptr<i32>>, tensor<32x16xi64> loc(#loc68)
+      %tmp0_25 = tt.broadcast %r0_mask_12 : tensor<1x16xi1> -> tensor<32x16xi1> loc(#loc69)
+      %tmp0_26 = tt.broadcast %xmask_4 : tensor<32x1xi1> -> tensor<32x16xi1> loc(#loc69)
+      %tmp0_27 = arith.andi %tmp0_25, %tmp0_26 : tensor<32x16xi1> loc(#loc69)
+      %tmp0_28 = tt.load %tmp0_24, %tmp0_27, %cst evictionPolicy = evict_last : tensor<32x16x!tt.ptr<i32>> loc(#loc70)
+      %tmp1 = arith.extsi %tmp0_28 : tensor<32x16xi32> to tensor<32x16xi64> loc(#loc71)
+      %tmp4 = arith.addi %_tmp3_10, %tmp1 : tensor<32x16xi64> loc(#loc72)
+      %_tmp3_29 = arith.select %tmp0_27, %tmp4, %_tmp3_10 : tensor<32x16xi1>, tensor<32x16xi64> loc(#loc73)
+      scf.yield %_tmp3_29 : tensor<32x16xi64> loc(#loc27)
+    } loc(#loc60)
+    %tmp3 = "tt.reduce"(%_tmp3_8) <{axis = 1 : i32}> ({
+    ^bb0(%tmp3_10: i64 loc(callsite(#loc1 at #loc74)), %tmp3_11: i64 loc(callsite(#loc1 at #loc74))):
+      %tmp3_12 = arith.addi %tmp3_10, %tmp3_11 : i64 loc(#loc80)
+      tt.reduce.return %tmp3_12 : i64 loc(#loc78)
+    }) : (tensor<32x16xi64>) -> tensor<32xi64> loc(#loc78)
+    %tmp3_9 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<32xi64> -> tensor<32x1xi64> loc(#loc75)
+    %tmp5 = arith.trunci %tmp3_9 : tensor<32x1xi64> to tensor<32x1xi32> loc(#loc76)
+    %0 = arith.cmpi sle, %ks0, %c1_i64 : i64 loc(#loc33)
+    %1 = arith.cmpi sgt, %ks0, %c1_i64 : i64 loc(#loc34)
+    %2 = arith.extui %1 : i1 to i64 loc(#loc35)
+    %3 = arith.muli %ks0, %2 : i64 loc(#loc35)
+    %4 = arith.extui %0 : i1 to i64 loc(#loc77)
+    %5 = arith.addi %4, %3 : i64 loc(#loc36)
+    %6 = tt.splat %5 : i64 -> tensor<32x1xi64> loc(#loc38)
+    %7 = arith.muli %x1, %6 : tensor<32x1xi64> loc(#loc38)
+    %8 = arith.addi %x0_7, %7 : tensor<32x1xi64> loc(#loc39)
+    %9 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<32x1x!tt.ptr<i32>> loc(#loc40)
+    %10 = tt.addptr %9, %8 : tensor<32x1x!tt.ptr<i32>>, tensor<32x1xi64> loc(#loc40)
+    tt.store %10, %tmp5, %xmask_4 : tensor<32x1x!tt.ptr<i32>> loc(#loc41)
+    tt.return loc(#loc42)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":30:40)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":28:43)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:28)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":21:33)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:36)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:44)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":22:23)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":23:21)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":24:27)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":24:37)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":26:19)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":27:19)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":31:31)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":32:29)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:43)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:39)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:54)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:58)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:50)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:34)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:73)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":36:63)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":37:23)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":39:23)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:48)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":40:8)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":41:28)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":42:19)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:49)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:75)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:66)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:57)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:41)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:34)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:30)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:25)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:88)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wc/cwc2jctz67s7n6fotl22gr44glijrcfxpeblvk32hornlllznx2m.py":43:4)
+#loc49 = loc("_tmp3"(#loc3))
+#loc50 = loc("xoffset"(#loc4))
+#loc51 = loc("xoffset"(#loc5))
+#loc52 = loc("xindex"(#loc6))
+#loc53 = loc("xindex"(#loc7))
+#loc54 = loc("xindex"(#loc8))
+#loc55 = loc("xmask"(#loc9))
+#loc56 = loc("r0_base"(#loc10))
+#loc57 = loc("r0_base"(#loc11))
+#loc58 = loc("x0"(#loc12))
+#loc59 = loc("x1"(#loc13))
+#loc60 = loc("_tmp3"(#loc2))
+#loc61 = loc("r0_index"(#loc14))
+#loc62 = loc("r0_mask"(#loc15))
+#loc63 = loc("tmp0"(#loc16))
+#loc64 = loc("tmp0"(#loc17))
+#loc65 = loc("tmp0"(#loc18))
+#loc66 = loc("tmp0"(#loc19))
+#loc67 = loc("tmp0"(#loc20))
+#loc68 = loc("tmp0"(#loc21))
+#loc69 = loc("tmp0"(#loc22))
+#loc70 = loc("tmp0"(#loc23))
+#loc71 = loc("tmp1"(#loc24))
+#loc72 = loc("tmp4"(#loc25))
+#loc73 = loc("_tmp3"(#loc26))
+#loc75 = loc("tmp3"(#loc31))
+#loc76 = loc("tmp5"(#loc32))
+#loc77 = loc(fused[#loc36, #loc37])
+#loc78 = loc(callsite(#loc28 at #loc74))
+#loc80 = loc(callsite(#loc30 at #loc78))

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/__grp__triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.source", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttir", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttgir", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.llir", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ptx", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.cubin", "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.json

@@ -0,0 +1 @@
+{"hash": "d4e9ec88be03aebb42041f53f2cbb9bb30afcd0e87a7a62c3ff7f5fe862eba44", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 4, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 0, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0"}

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.llir

@@ -0,0 +1,667 @@
+; ModuleID = 'LLVMDialectModule'
+source_filename = "LLVMDialectModule"
+target datalayout = "e-p3:32:32-p4:32:32-p5:32:32-p6:32:32-p7:32:32-i64:64-i128:128-v16:16-v32:32-n16:32:64"
+
+@assertFunc_1 = internal constant [8 x i8] c"unknown\00"
+@assertFile_1 = internal constant [114 x i8] c"/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py\00"
+@assertMessage_1 = internal constant [38 x i8] c"index out of bounds: 0 <= tmp25 < ks4\00"
+@assertFunc_0 = internal constant [8 x i8] c"unknown\00"
+@assertFile_0 = internal constant [114 x i8] c"/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py\00"
+@assertMessage_0 = internal constant [37 x i8] c"index out of bounds: 0 <= tmp5 < ks2\00"
+
+; Function Attrs: noreturn
+declare !dbg !5 void @__assertfail(ptr, ptr, i32, ptr, i64) local_unnamed_addr #0
+
+define ptx_kernel void @triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0(ptr addrspace(1) %0, ptr addrspace(1) %1, ptr addrspace(1) %2, ptr addrspace(1) %3, ptr addrspace(1) %4, i64 %5, i64 %6, i64 %7, i64 %8, i64 %9, i32 %10, ptr addrspace(1) readnone captures(none) %11, ptr addrspace(1) readnone captures(none) %12) local_unnamed_addr #1 !dbg !9 {
+  %14 = tail call i32 @llvm.nvvm.read.ptx.sreg.ctaid.x(), !dbg !10
+  %15 = shl i32 %14, 10, !dbg !11
+  %16 = tail call i32 @llvm.nvvm.read.ptx.sreg.tid.x(), !dbg !12
+  %17 = shl nuw nsw i32 %16, 3, !dbg !12
+  %18 = and i32 %17, 1016, !dbg !12
+  %19 = or disjoint i32 %18, %15, !dbg !13
+  %20 = or disjoint i32 %19, 1, !dbg !13
+  %21 = or disjoint i32 %19, 2, !dbg !13
+  %22 = or disjoint i32 %19, 3, !dbg !13
+  %23 = or disjoint i32 %19, 4, !dbg !13
+  %24 = or disjoint i32 %19, 5, !dbg !13
+  %25 = or disjoint i32 %19, 6, !dbg !13
+  %26 = or disjoint i32 %19, 7, !dbg !13
+  %27 = insertelement <8 x i32> poison, i32 %26, i64 0, !dbg !14
+  %28 = insertelement <8 x i32> %27, i32 %25, i64 1, !dbg !14
+  %29 = insertelement <8 x i32> %28, i32 %24, i64 2, !dbg !14
+  %30 = insertelement <8 x i32> %29, i32 %23, i64 3, !dbg !14
+  %31 = insertelement <8 x i32> %30, i32 %22, i64 4, !dbg !14
+  %32 = insertelement <8 x i32> %31, i32 %21, i64 5, !dbg !14
+  %33 = insertelement <8 x i32> %32, i32 %20, i64 6, !dbg !14
+  %34 = insertelement <8 x i32> %33, i32 %19, i64 7, !dbg !14
+  %35 = sext <8 x i32> %34 to <8 x i64>, !dbg !14
+  %36 = extractelement <8 x i64> %35, i64 7, !dbg !15
+  %37 = sdiv i64 %36, %5, !dbg !14
+  %38 = extractelement <8 x i64> %35, i64 6, !dbg !15
+  %39 = sdiv i64 %38, %5, !dbg !14
+  %40 = extractelement <8 x i64> %35, i64 5, !dbg !15
+  %41 = sdiv i64 %40, %5, !dbg !14
+  %42 = extractelement <8 x i64> %35, i64 4, !dbg !15
+  %43 = sdiv i64 %42, %5, !dbg !14
+  %44 = extractelement <8 x i64> %35, i64 3, !dbg !15
+  %45 = sdiv i64 %44, %5, !dbg !14
+  %46 = extractelement <8 x i64> %35, i64 2, !dbg !15
+  %47 = sdiv i64 %46, %5, !dbg !14
+  %48 = extractelement <8 x i64> %35, i64 1, !dbg !15
+  %49 = sdiv i64 %48, %5, !dbg !14
+  %50 = extractelement <8 x i64> %35, i64 0, !dbg !15
+  %51 = sdiv i64 %50, %5, !dbg !14
+  %52 = srem i64 %37, %6, !dbg !16
+  %53 = srem i64 %39, %6, !dbg !16
+  %54 = srem i64 %41, %6, !dbg !16
+  %55 = srem i64 %43, %6, !dbg !16
+  %56 = srem i64 %45, %6, !dbg !16
+  %57 = srem i64 %47, %6, !dbg !16
+  %58 = srem i64 %49, %6, !dbg !16
+  %59 = srem i64 %51, %6, !dbg !16
+  %60 = getelementptr bfloat, ptr addrspace(1) %0, i64 %36, !dbg !15
+  %61 = getelementptr bfloat, ptr addrspace(1) %0, i64 %38, !dbg !15
+  %62 = getelementptr bfloat, ptr addrspace(1) %0, i64 %40, !dbg !15
+  %63 = getelementptr bfloat, ptr addrspace(1) %0, i64 %42, !dbg !15
+  %64 = getelementptr bfloat, ptr addrspace(1) %0, i64 %44, !dbg !15
+  %65 = getelementptr bfloat, ptr addrspace(1) %0, i64 %46, !dbg !15
+  %66 = getelementptr bfloat, ptr addrspace(1) %0, i64 %48, !dbg !15
+  %67 = getelementptr bfloat, ptr addrspace(1) %0, i64 %50, !dbg !15
+  %68 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %69 = getelementptr i64, ptr addrspace(1) %1, i64 %52, !dbg !18
+  %70 = getelementptr i64, ptr addrspace(1) %1, i64 %53, !dbg !18
+  %71 = getelementptr i64, ptr addrspace(1) %1, i64 %54, !dbg !18
+  %72 = getelementptr i64, ptr addrspace(1) %1, i64 %55, !dbg !18
+  %73 = getelementptr i64, ptr addrspace(1) %1, i64 %56, !dbg !18
+  %74 = getelementptr i64, ptr addrspace(1) %1, i64 %57, !dbg !18
+  %75 = getelementptr i64, ptr addrspace(1) %1, i64 %58, !dbg !18
+  %76 = getelementptr i64, ptr addrspace(1) %1, i64 %59, !dbg !18
+  %77 = insertelement <8 x i32> poison, i32 %19, i64 0, !dbg !19
+  %78 = insertelement <8 x i32> %77, i32 %20, i64 1, !dbg !19
+  %79 = insertelement <8 x i32> %78, i32 %21, i64 2, !dbg !19
+  %80 = insertelement <8 x i32> %79, i32 %22, i64 3, !dbg !19
+  %81 = insertelement <8 x i32> %80, i32 %23, i64 4, !dbg !19
+  %82 = insertelement <8 x i32> %81, i32 %24, i64 5, !dbg !19
+  %83 = insertelement <8 x i32> %82, i32 %25, i64 6, !dbg !19
+  %84 = insertelement <8 x i32> %83, i32 %26, i64 7, !dbg !19
+  %85 = insertelement <8 x i32> poison, i32 %10, i64 0, !dbg !19
+  %86 = shufflevector <8 x i32> %85, <8 x i32> poison, <8 x i32> zeroinitializer, !dbg !19
+  %87 = icmp slt <8 x i32> %84, %86, !dbg !19
+  %88 = extractelement <8 x i1> %87, i64 0, !dbg !17
+  %89 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %60, i64 %68, i1 %88) #4, !dbg !17
+  %90 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %91 = extractelement <8 x i1> %87, i64 1, !dbg !17
+  %92 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %61, i64 %90, i1 %91) #4, !dbg !17
+  %93 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %94 = extractelement <8 x i1> %87, i64 2, !dbg !17
+  %95 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %62, i64 %93, i1 %94) #4, !dbg !17
+  %96 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %97 = extractelement <8 x i1> %87, i64 3, !dbg !17
+  %98 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %63, i64 %96, i1 %97) #4, !dbg !17
+  %99 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %100 = extractelement <8 x i1> %87, i64 4, !dbg !17
+  %101 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %64, i64 %99, i1 %100) #4, !dbg !17
+  %102 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %103 = extractelement <8 x i1> %87, i64 5, !dbg !17
+  %104 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %65, i64 %102, i1 %103) #4, !dbg !17
+  %105 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %106 = extractelement <8 x i1> %87, i64 6, !dbg !17
+  %107 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %66, i64 %105, i1 %106) #4, !dbg !17
+  %108 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !17
+  %109 = extractelement <8 x i1> %87, i64 7, !dbg !17
+  %110 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %67, i64 %108, i1 %109) #4, !dbg !17
+  %111 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %112 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %69, i64 %111, i1 %88) #4, !dbg !20
+  %113 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %114 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %70, i64 %113, i1 %91) #4, !dbg !20
+  %115 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %116 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %71, i64 %115, i1 %94) #4, !dbg !20
+  %117 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %118 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %72, i64 %117, i1 %97) #4, !dbg !20
+  %119 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %120 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %73, i64 %119, i1 %100) #4, !dbg !20
+  %121 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %122 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %74, i64 %121, i1 %103) #4, !dbg !20
+  %123 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %124 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %75, i64 %123, i1 %106) #4, !dbg !20
+  %125 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !20
+  %126 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b64 { $0 }, [ $1 + 0 ], $2;", "=l,l,l,b"(ptr addrspace(1) %76, i64 %125, i1 %109) #4, !dbg !20
+  %127 = insertelement <8 x i64> poison, i64 %112, i64 0, !dbg !21
+  %128 = insertelement <8 x i64> %127, i64 %114, i64 1, !dbg !21
+  %129 = insertelement <8 x i64> %128, i64 %116, i64 2, !dbg !21
+  %130 = insertelement <8 x i64> %129, i64 %118, i64 3, !dbg !21
+  %131 = insertelement <8 x i64> %130, i64 %120, i64 4, !dbg !21
+  %132 = insertelement <8 x i64> %131, i64 %122, i64 5, !dbg !21
+  %133 = insertelement <8 x i64> %132, i64 %124, i64 6, !dbg !21
+  %134 = insertelement <8 x i64> %133, i64 %126, i64 7, !dbg !21
+  %135 = icmp slt <8 x i64> %134, zeroinitializer, !dbg !21
+  %136 = insertelement <8 x i64> poison, i64 %7, i64 0, !dbg !22
+  %137 = shufflevector <8 x i64> %136, <8 x i64> poison, <8 x i32> zeroinitializer, !dbg !22
+  %138 = select <8 x i1> %135, <8 x i64> %137, <8 x i64> zeroinitializer, !dbg !22
+  %139 = add <8 x i64> %138, %134, !dbg !22
+  %140 = icmp slt <8 x i64> %139, zeroinitializer, !dbg !23
+  %141 = icmp sge <8 x i64> %139, %137, !dbg !24
+  %142 = or <8 x i1> %140, %141, !dbg !25
+  %143 = and <8 x i1> %87, %142, !dbg !26
+  %144 = bitcast <8 x i1> %143 to i8, !dbg !27
+  %.not = icmp eq i8 %144, 0, !dbg !27
+  br i1 %.not, label %146, label %145, !dbg !27
+
+145:                                              ; preds = %13
+  tail call void @__assertfail(ptr nonnull @assertMessage_0, ptr nonnull @assertFile_0, i32 32, ptr nonnull @assertFunc_0, i64 1), !dbg !27
+  unreachable, !dbg !27
+
+146:                                              ; preds = %13
+  %147 = insertelement <8 x i64> poison, i64 %8, i64 0, !dbg !28
+  %148 = shufflevector <8 x i64> %147, <8 x i64> poison, <8 x i32> zeroinitializer, !dbg !28
+  %149 = srem <8 x i64> %35, %148, !dbg !28
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !27
+  %150 = extractelement <8 x i64> %139, i64 0, !dbg !29
+  %151 = mul i64 %150, %8, !dbg !29
+  %152 = extractelement <8 x i64> %139, i64 1, !dbg !29
+  %153 = mul i64 %152, %8, !dbg !29
+  %154 = extractelement <8 x i64> %139, i64 2, !dbg !29
+  %155 = mul i64 %154, %8, !dbg !29
+  %156 = extractelement <8 x i64> %139, i64 3, !dbg !29
+  %157 = mul i64 %156, %8, !dbg !29
+  %158 = extractelement <8 x i64> %139, i64 4, !dbg !29
+  %159 = mul i64 %158, %8, !dbg !29
+  %160 = extractelement <8 x i64> %139, i64 5, !dbg !29
+  %161 = mul i64 %160, %8, !dbg !29
+  %162 = extractelement <8 x i64> %139, i64 6, !dbg !29
+  %163 = mul i64 %162, %8, !dbg !29
+  %164 = extractelement <8 x i64> %139, i64 7, !dbg !29
+  %165 = mul i64 %164, %8, !dbg !29
+  %166 = extractelement <8 x i64> %149, i64 7, !dbg !30
+  %167 = getelementptr bfloat, ptr addrspace(1) %2, i64 %166, !dbg !31
+  %168 = getelementptr bfloat, ptr addrspace(1) %167, i64 %151, !dbg !31
+  %169 = extractelement <8 x i64> %149, i64 6, !dbg !30
+  %170 = getelementptr bfloat, ptr addrspace(1) %2, i64 %169, !dbg !31
+  %171 = getelementptr bfloat, ptr addrspace(1) %170, i64 %153, !dbg !31
+  %172 = extractelement <8 x i64> %149, i64 5, !dbg !30
+  %173 = getelementptr bfloat, ptr addrspace(1) %2, i64 %172, !dbg !31
+  %174 = getelementptr bfloat, ptr addrspace(1) %173, i64 %155, !dbg !31
+  %175 = extractelement <8 x i64> %149, i64 4, !dbg !30
+  %176 = getelementptr bfloat, ptr addrspace(1) %2, i64 %175, !dbg !31
+  %177 = getelementptr bfloat, ptr addrspace(1) %176, i64 %157, !dbg !31
+  %178 = extractelement <8 x i64> %149, i64 3, !dbg !30
+  %179 = getelementptr bfloat, ptr addrspace(1) %2, i64 %178, !dbg !31
+  %180 = getelementptr bfloat, ptr addrspace(1) %179, i64 %159, !dbg !31
+  %181 = extractelement <8 x i64> %149, i64 2, !dbg !30
+  %182 = getelementptr bfloat, ptr addrspace(1) %2, i64 %181, !dbg !31
+  %183 = getelementptr bfloat, ptr addrspace(1) %182, i64 %161, !dbg !31
+  %184 = extractelement <8 x i64> %149, i64 1, !dbg !30
+  %185 = getelementptr bfloat, ptr addrspace(1) %2, i64 %184, !dbg !31
+  %186 = getelementptr bfloat, ptr addrspace(1) %185, i64 %163, !dbg !31
+  %187 = extractelement <8 x i64> %149, i64 0, !dbg !30
+  %188 = getelementptr bfloat, ptr addrspace(1) %2, i64 %187, !dbg !31
+  %189 = getelementptr bfloat, ptr addrspace(1) %188, i64 %165, !dbg !31
+  %190 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %191 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %168, i64 %190, i1 %88) #4, !dbg !32
+  %192 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %193 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %171, i64 %192, i1 %91) #4, !dbg !32
+  %194 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %195 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %174, i64 %194, i1 %94) #4, !dbg !32
+  %196 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %197 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %177, i64 %196, i1 %97) #4, !dbg !32
+  %198 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %199 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %180, i64 %198, i1 %100) #4, !dbg !32
+  %200 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %201 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %183, i64 %200, i1 %103) #4, !dbg !32
+  %202 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %203 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %186, i64 %202, i1 %106) #4, !dbg !32
+  %204 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !32
+  %205 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %189, i64 %204, i1 %109) #4, !dbg !32
+  %206 = sdiv i64 %8, 2, !dbg !33
+  %207 = sub i64 %8, %206, !dbg !34
+  %208 = icmp slt i64 %166, %207, !dbg !35
+  %209 = icmp slt i64 %169, %207, !dbg !35
+  %210 = icmp slt i64 %172, %207, !dbg !35
+  %211 = icmp slt i64 %175, %207, !dbg !35
+  %212 = icmp slt i64 %178, %207, !dbg !35
+  %213 = icmp slt i64 %181, %207, !dbg !35
+  %214 = icmp slt i64 %184, %207, !dbg !35
+  %215 = icmp slt i64 %187, %207, !dbg !35
+  %216 = sub nsw i64 %36, %166, !dbg !30
+  %217 = sub nsw i64 %38, %169, !dbg !30
+  %218 = sub nsw i64 %40, %172, !dbg !30
+  %219 = sub nsw i64 %42, %175, !dbg !30
+  %220 = sub nsw i64 %44, %178, !dbg !30
+  %221 = sub nsw i64 %46, %181, !dbg !30
+  %222 = sub nsw i64 %48, %184, !dbg !30
+  %223 = sub nsw i64 %50, %187, !dbg !30
+  %224 = getelementptr bfloat, ptr addrspace(1) %0, i64 %216, !dbg !36
+  %225 = getelementptr bfloat, ptr addrspace(1) %224, i64 %206, !dbg !36
+  %226 = getelementptr bfloat, ptr addrspace(1) %225, i64 %166, !dbg !36
+  %227 = getelementptr bfloat, ptr addrspace(1) %0, i64 %217, !dbg !36
+  %228 = getelementptr bfloat, ptr addrspace(1) %227, i64 %206, !dbg !36
+  %229 = getelementptr bfloat, ptr addrspace(1) %228, i64 %169, !dbg !36
+  %230 = getelementptr bfloat, ptr addrspace(1) %0, i64 %218, !dbg !36
+  %231 = getelementptr bfloat, ptr addrspace(1) %230, i64 %206, !dbg !36
+  %232 = getelementptr bfloat, ptr addrspace(1) %231, i64 %172, !dbg !36
+  %233 = getelementptr bfloat, ptr addrspace(1) %0, i64 %219, !dbg !36
+  %234 = getelementptr bfloat, ptr addrspace(1) %233, i64 %206, !dbg !36
+  %235 = getelementptr bfloat, ptr addrspace(1) %234, i64 %175, !dbg !36
+  %236 = getelementptr bfloat, ptr addrspace(1) %0, i64 %220, !dbg !36
+  %237 = getelementptr bfloat, ptr addrspace(1) %236, i64 %206, !dbg !36
+  %238 = getelementptr bfloat, ptr addrspace(1) %237, i64 %178, !dbg !36
+  %239 = getelementptr bfloat, ptr addrspace(1) %0, i64 %221, !dbg !36
+  %240 = getelementptr bfloat, ptr addrspace(1) %239, i64 %206, !dbg !36
+  %241 = getelementptr bfloat, ptr addrspace(1) %240, i64 %181, !dbg !36
+  %242 = getelementptr bfloat, ptr addrspace(1) %0, i64 %222, !dbg !36
+  %243 = getelementptr bfloat, ptr addrspace(1) %242, i64 %206, !dbg !36
+  %244 = getelementptr bfloat, ptr addrspace(1) %243, i64 %184, !dbg !36
+  %245 = getelementptr bfloat, ptr addrspace(1) %0, i64 %223, !dbg !36
+  %246 = getelementptr bfloat, ptr addrspace(1) %245, i64 %206, !dbg !36
+  %247 = getelementptr bfloat, ptr addrspace(1) %246, i64 %187, !dbg !36
+  %248 = and i1 %88, %208, !dbg !37
+  %249 = and i1 %91, %209, !dbg !37
+  %250 = and i1 %94, %210, !dbg !37
+  %251 = and i1 %97, %211, !dbg !37
+  %252 = and i1 %100, %212, !dbg !37
+  %253 = and i1 %103, %213, !dbg !37
+  %254 = and i1 %106, %214, !dbg !37
+  %255 = and i1 %109, %215, !dbg !37
+  %256 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %257 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %226, i64 %256, i1 %248) #4, !dbg !38
+  %258 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %259 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %229, i64 %258, i1 %249) #4, !dbg !38
+  %260 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %261 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %232, i64 %260, i1 %250) #4, !dbg !38
+  %262 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %263 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %235, i64 %262, i1 %251) #4, !dbg !38
+  %264 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %265 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %238, i64 %264, i1 %252) #4, !dbg !38
+  %266 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %267 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %241, i64 %266, i1 %253) #4, !dbg !38
+  %268 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %269 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %244, i64 %268, i1 %254) #4, !dbg !38
+  %270 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !38
+  %271 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %247, i64 %270, i1 %255) #4, !dbg !38
+  %272 = insertelement <8 x i64> poison, i64 %207, i64 0, !dbg !39
+  %273 = shufflevector <8 x i64> %272, <8 x i64> poison, <8 x i32> zeroinitializer, !dbg !39
+  %274 = icmp sge <8 x i64> %149, %273, !dbg !39
+  %275 = sub i64 %166, %8, !dbg !40
+  %276 = sub i64 %169, %8, !dbg !40
+  %277 = sub i64 %172, %8, !dbg !40
+  %278 = sub i64 %175, %8, !dbg !40
+  %279 = sub i64 %178, %8, !dbg !40
+  %280 = sub i64 %181, %8, !dbg !40
+  %281 = sub i64 %184, %8, !dbg !40
+  %282 = sub i64 %187, %8, !dbg !40
+  %283 = getelementptr bfloat, ptr addrspace(1) %224, i64 %275, !dbg !41
+  %284 = getelementptr bfloat, ptr addrspace(1) %283, i64 %206, !dbg !41
+  %285 = getelementptr bfloat, ptr addrspace(1) %227, i64 %276, !dbg !41
+  %286 = getelementptr bfloat, ptr addrspace(1) %285, i64 %206, !dbg !41
+  %287 = getelementptr bfloat, ptr addrspace(1) %230, i64 %277, !dbg !41
+  %288 = getelementptr bfloat, ptr addrspace(1) %287, i64 %206, !dbg !41
+  %289 = getelementptr bfloat, ptr addrspace(1) %233, i64 %278, !dbg !41
+  %290 = getelementptr bfloat, ptr addrspace(1) %289, i64 %206, !dbg !41
+  %291 = getelementptr bfloat, ptr addrspace(1) %236, i64 %279, !dbg !41
+  %292 = getelementptr bfloat, ptr addrspace(1) %291, i64 %206, !dbg !41
+  %293 = getelementptr bfloat, ptr addrspace(1) %239, i64 %280, !dbg !41
+  %294 = getelementptr bfloat, ptr addrspace(1) %293, i64 %206, !dbg !41
+  %295 = getelementptr bfloat, ptr addrspace(1) %242, i64 %281, !dbg !41
+  %296 = getelementptr bfloat, ptr addrspace(1) %295, i64 %206, !dbg !41
+  %297 = getelementptr bfloat, ptr addrspace(1) %245, i64 %282, !dbg !41
+  %298 = getelementptr bfloat, ptr addrspace(1) %297, i64 %206, !dbg !41
+  %299 = extractelement <8 x i1> %274, i64 7, !dbg !42
+  %300 = and i1 %88, %299, !dbg !42
+  %301 = extractelement <8 x i1> %274, i64 6, !dbg !42
+  %302 = and i1 %91, %301, !dbg !42
+  %303 = extractelement <8 x i1> %274, i64 5, !dbg !42
+  %304 = and i1 %94, %303, !dbg !42
+  %305 = extractelement <8 x i1> %274, i64 4, !dbg !42
+  %306 = and i1 %97, %305, !dbg !42
+  %307 = extractelement <8 x i1> %274, i64 3, !dbg !42
+  %308 = and i1 %100, %307, !dbg !42
+  %309 = extractelement <8 x i1> %274, i64 2, !dbg !42
+  %310 = and i1 %103, %309, !dbg !42
+  %311 = extractelement <8 x i1> %274, i64 1, !dbg !42
+  %312 = and i1 %106, %311, !dbg !42
+  %313 = extractelement <8 x i1> %274, i64 0, !dbg !42
+  %314 = and i1 %109, %313, !dbg !42
+  %315 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %316 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %284, i64 %315, i1 %300) #4, !dbg !43
+  %317 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %318 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %286, i64 %317, i1 %302) #4, !dbg !43
+  %319 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %320 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %288, i64 %319, i1 %304) #4, !dbg !43
+  %321 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %322 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %290, i64 %321, i1 %306) #4, !dbg !43
+  %323 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %324 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %292, i64 %323, i1 %308) #4, !dbg !43
+  %325 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %326 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %294, i64 %325, i1 %310) #4, !dbg !43
+  %327 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %328 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %296, i64 %327, i1 %312) #4, !dbg !43
+  %329 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !43
+  %330 = tail call i16 asm sideeffect "mov.u16 $0, $1;\0A\09@$4 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $2 + 0 ], $3;", "=c,c,l,l,b"(i16 0, ptr addrspace(1) %298, i64 %329, i1 %314) #4, !dbg !43
+  %331 = insertelement <8 x i64> poison, i64 %9, i64 0, !dbg !44
+  %332 = shufflevector <8 x i64> %331, <8 x i64> poison, <8 x i32> zeroinitializer, !dbg !44
+  %333 = select <8 x i1> %135, <8 x i64> %332, <8 x i64> zeroinitializer, !dbg !44
+  %334 = add <8 x i64> %333, %134, !dbg !44
+  %335 = icmp slt <8 x i64> %334, zeroinitializer, !dbg !45
+  %336 = icmp sge <8 x i64> %334, %332, !dbg !46
+  %337 = or <8 x i1> %335, %336, !dbg !47
+  %338 = and <8 x i1> %87, %337, !dbg !48
+  %339 = bitcast <8 x i1> %338 to i8, !dbg !49
+  %.not87 = icmp eq i8 %339, 0, !dbg !49
+  br i1 %.not87, label %341, label %340, !dbg !49
+
+340:                                              ; preds = %146
+  tail call void @__assertfail(ptr nonnull @assertMessage_1, ptr nonnull @assertFile_1, i32 52, ptr nonnull @assertFunc_1, i64 1), !dbg !49
+  unreachable, !dbg !49
+
+341:                                              ; preds = %146
+  %342 = bitcast i16 %271 to bfloat, !dbg !38
+  %343 = fpext bfloat %342 to float, !dbg !50
+  %344 = fsub float 0.000000e+00, %343, !dbg !51
+  %345 = bitcast i16 %330 to bfloat, !dbg !43
+  %346 = fpext bfloat %345 to float, !dbg !52
+  %347 = select i1 %215, float %344, float %346, !dbg !53
+  %348 = bitcast i16 %269 to bfloat, !dbg !38
+  %349 = fpext bfloat %348 to float, !dbg !50
+  %350 = fsub float 0.000000e+00, %349, !dbg !51
+  %351 = bitcast i16 %328 to bfloat, !dbg !43
+  %352 = fpext bfloat %351 to float, !dbg !52
+  %353 = select i1 %214, float %350, float %352, !dbg !53
+  %354 = bitcast i16 %267 to bfloat, !dbg !38
+  %355 = fpext bfloat %354 to float, !dbg !50
+  %356 = fsub float 0.000000e+00, %355, !dbg !51
+  %357 = bitcast i16 %326 to bfloat, !dbg !43
+  %358 = fpext bfloat %357 to float, !dbg !52
+  %359 = select i1 %213, float %356, float %358, !dbg !53
+  %360 = bitcast i16 %265 to bfloat, !dbg !38
+  %361 = fpext bfloat %360 to float, !dbg !50
+  %362 = fsub float 0.000000e+00, %361, !dbg !51
+  %363 = bitcast i16 %324 to bfloat, !dbg !43
+  %364 = fpext bfloat %363 to float, !dbg !52
+  %365 = select i1 %212, float %362, float %364, !dbg !53
+  %366 = bitcast i16 %263 to bfloat, !dbg !38
+  %367 = fpext bfloat %366 to float, !dbg !50
+  %368 = fsub float 0.000000e+00, %367, !dbg !51
+  %369 = bitcast i16 %322 to bfloat, !dbg !43
+  %370 = fpext bfloat %369 to float, !dbg !52
+  %371 = select i1 %211, float %368, float %370, !dbg !53
+  %372 = bitcast i16 %261 to bfloat, !dbg !38
+  %373 = fpext bfloat %372 to float, !dbg !50
+  %374 = fsub float 0.000000e+00, %373, !dbg !51
+  %375 = bitcast i16 %320 to bfloat, !dbg !43
+  %376 = fpext bfloat %375 to float, !dbg !52
+  %377 = select i1 %210, float %374, float %376, !dbg !53
+  %378 = bitcast i16 %259 to bfloat, !dbg !38
+  %379 = fpext bfloat %378 to float, !dbg !50
+  %380 = fsub float 0.000000e+00, %379, !dbg !51
+  %381 = bitcast i16 %318 to bfloat, !dbg !43
+  %382 = fpext bfloat %381 to float, !dbg !52
+  %383 = select i1 %209, float %380, float %382, !dbg !53
+  %384 = bitcast i16 %257 to bfloat, !dbg !38
+  %385 = fpext bfloat %384 to float, !dbg !50
+  %386 = fsub float 0.000000e+00, %385, !dbg !51
+  %387 = bitcast i16 %316 to bfloat, !dbg !43
+  %388 = fpext bfloat %387 to float, !dbg !52
+  %389 = select i1 %208, float %386, float %388, !dbg !53
+  %390 = bitcast i16 %110 to bfloat, !dbg !17
+  %391 = fpext bfloat %390 to float, !dbg !54
+  %392 = bitcast i16 %107 to bfloat, !dbg !17
+  %393 = fpext bfloat %392 to float, !dbg !54
+  %394 = bitcast i16 %104 to bfloat, !dbg !17
+  %395 = fpext bfloat %394 to float, !dbg !54
+  %396 = bitcast i16 %101 to bfloat, !dbg !17
+  %397 = fpext bfloat %396 to float, !dbg !54
+  %398 = bitcast i16 %98 to bfloat, !dbg !17
+  %399 = fpext bfloat %398 to float, !dbg !54
+  %400 = bitcast i16 %95 to bfloat, !dbg !17
+  %401 = fpext bfloat %400 to float, !dbg !54
+  %402 = bitcast i16 %92 to bfloat, !dbg !17
+  %403 = fpext bfloat %402 to float, !dbg !54
+  %404 = bitcast i16 %89 to bfloat, !dbg !17
+  %405 = fpext bfloat %404 to float, !dbg !54
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !49
+  %406 = extractelement <8 x i64> %334, i64 0, !dbg !55
+  %407 = mul i64 %406, %8, !dbg !55
+  %408 = extractelement <8 x i64> %334, i64 1, !dbg !55
+  %409 = mul i64 %408, %8, !dbg !55
+  %410 = extractelement <8 x i64> %334, i64 2, !dbg !55
+  %411 = mul i64 %410, %8, !dbg !55
+  %412 = extractelement <8 x i64> %334, i64 3, !dbg !55
+  %413 = mul i64 %412, %8, !dbg !55
+  %414 = extractelement <8 x i64> %334, i64 4, !dbg !55
+  %415 = mul i64 %414, %8, !dbg !55
+  %416 = extractelement <8 x i64> %334, i64 5, !dbg !55
+  %417 = mul i64 %416, %8, !dbg !55
+  %418 = extractelement <8 x i64> %334, i64 6, !dbg !55
+  %419 = mul i64 %418, %8, !dbg !55
+  %420 = extractelement <8 x i64> %334, i64 7, !dbg !55
+  %421 = mul i64 %420, %8, !dbg !55
+  %422 = getelementptr bfloat, ptr addrspace(1) %3, i64 %166, !dbg !56
+  %423 = getelementptr bfloat, ptr addrspace(1) %422, i64 %407, !dbg !56
+  %424 = getelementptr bfloat, ptr addrspace(1) %3, i64 %169, !dbg !56
+  %425 = getelementptr bfloat, ptr addrspace(1) %424, i64 %409, !dbg !56
+  %426 = getelementptr bfloat, ptr addrspace(1) %3, i64 %172, !dbg !56
+  %427 = getelementptr bfloat, ptr addrspace(1) %426, i64 %411, !dbg !56
+  %428 = getelementptr bfloat, ptr addrspace(1) %3, i64 %175, !dbg !56
+  %429 = getelementptr bfloat, ptr addrspace(1) %428, i64 %413, !dbg !56
+  %430 = getelementptr bfloat, ptr addrspace(1) %3, i64 %178, !dbg !56
+  %431 = getelementptr bfloat, ptr addrspace(1) %430, i64 %415, !dbg !56
+  %432 = getelementptr bfloat, ptr addrspace(1) %3, i64 %181, !dbg !56
+  %433 = getelementptr bfloat, ptr addrspace(1) %432, i64 %417, !dbg !56
+  %434 = getelementptr bfloat, ptr addrspace(1) %3, i64 %184, !dbg !56
+  %435 = getelementptr bfloat, ptr addrspace(1) %434, i64 %419, !dbg !56
+  %436 = getelementptr bfloat, ptr addrspace(1) %3, i64 %187, !dbg !56
+  %437 = getelementptr bfloat, ptr addrspace(1) %436, i64 %421, !dbg !56
+  %438 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %439 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %423, i64 %438, i1 %88) #4, !dbg !57
+  %440 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %441 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %425, i64 %440, i1 %91) #4, !dbg !57
+  %442 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %443 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %427, i64 %442, i1 %94) #4, !dbg !57
+  %444 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %445 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %429, i64 %444, i1 %97) #4, !dbg !57
+  %446 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %447 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %431, i64 %446, i1 %100) #4, !dbg !57
+  %448 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %449 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %433, i64 %448, i1 %103) #4, !dbg !57
+  %450 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %451 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %435, i64 %450, i1 %106) #4, !dbg !57
+  %452 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #4, !dbg !57
+  %453 = tail call i16 asm sideeffect "mov.u16 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b16 { $0 }, [ $1 + 0 ], $2;", "=c,l,l,b"(ptr addrspace(1) %437, i64 %452, i1 %109) #4, !dbg !57
+  %454 = insertelement <2 x i16> poison, i16 %191, i64 0, !dbg !32
+  %455 = insertelement <2 x i16> %454, i16 %439, i64 1, !dbg !32
+  %456 = bitcast <2 x i16> %455 to <2 x bfloat>, !dbg !32
+  %457 = fpext <2 x bfloat> %456 to <2 x float>, !dbg !58
+  %458 = insertelement <2 x float> poison, float %405, i64 0, !dbg !59
+  %459 = insertelement <2 x float> %458, float %389, i64 1, !dbg !59
+  %460 = fmul <2 x float> %459, %457, !dbg !59
+  %461 = insertelement <2 x i16> poison, i16 %193, i64 0, !dbg !32
+  %462 = insertelement <2 x i16> %461, i16 %441, i64 1, !dbg !32
+  %463 = bitcast <2 x i16> %462 to <2 x bfloat>, !dbg !32
+  %464 = fpext <2 x bfloat> %463 to <2 x float>, !dbg !58
+  %465 = insertelement <2 x float> poison, float %403, i64 0, !dbg !59
+  %466 = insertelement <2 x float> %465, float %383, i64 1, !dbg !59
+  %467 = fmul <2 x float> %466, %464, !dbg !59
+  %468 = insertelement <2 x i16> poison, i16 %195, i64 0, !dbg !32
+  %469 = insertelement <2 x i16> %468, i16 %443, i64 1, !dbg !32
+  %470 = bitcast <2 x i16> %469 to <2 x bfloat>, !dbg !32
+  %471 = fpext <2 x bfloat> %470 to <2 x float>, !dbg !58
+  %472 = insertelement <2 x float> poison, float %401, i64 0, !dbg !59
+  %473 = insertelement <2 x float> %472, float %377, i64 1, !dbg !59
+  %474 = fmul <2 x float> %473, %471, !dbg !59
+  %475 = insertelement <2 x i16> poison, i16 %197, i64 0, !dbg !32
+  %476 = insertelement <2 x i16> %475, i16 %445, i64 1, !dbg !32
+  %477 = bitcast <2 x i16> %476 to <2 x bfloat>, !dbg !32
+  %478 = fpext <2 x bfloat> %477 to <2 x float>, !dbg !58
+  %479 = insertelement <2 x float> poison, float %399, i64 0, !dbg !59
+  %480 = insertelement <2 x float> %479, float %371, i64 1, !dbg !59
+  %481 = fmul <2 x float> %480, %478, !dbg !59
+  %482 = insertelement <2 x i16> poison, i16 %199, i64 0, !dbg !32
+  %483 = insertelement <2 x i16> %482, i16 %447, i64 1, !dbg !32
+  %484 = bitcast <2 x i16> %483 to <2 x bfloat>, !dbg !32
+  %485 = fpext <2 x bfloat> %484 to <2 x float>, !dbg !58
+  %486 = insertelement <2 x float> poison, float %397, i64 0, !dbg !59
+  %487 = insertelement <2 x float> %486, float %365, i64 1, !dbg !59
+  %488 = fmul <2 x float> %487, %485, !dbg !59
+  %489 = insertelement <2 x i16> poison, i16 %201, i64 0, !dbg !32
+  %490 = insertelement <2 x i16> %489, i16 %449, i64 1, !dbg !32
+  %491 = bitcast <2 x i16> %490 to <2 x bfloat>, !dbg !32
+  %492 = fpext <2 x bfloat> %491 to <2 x float>, !dbg !58
+  %493 = insertelement <2 x float> poison, float %395, i64 0, !dbg !59
+  %494 = insertelement <2 x float> %493, float %359, i64 1, !dbg !59
+  %495 = fmul <2 x float> %494, %492, !dbg !59
+  %496 = insertelement <2 x i16> poison, i16 %203, i64 0, !dbg !32
+  %497 = insertelement <2 x i16> %496, i16 %451, i64 1, !dbg !32
+  %498 = bitcast <2 x i16> %497 to <2 x bfloat>, !dbg !32
+  %499 = fpext <2 x bfloat> %498 to <2 x float>, !dbg !58
+  %500 = insertelement <2 x float> poison, float %393, i64 0, !dbg !59
+  %501 = insertelement <2 x float> %500, float %353, i64 1, !dbg !59
+  %502 = fmul <2 x float> %501, %499, !dbg !59
+  %503 = insertelement <2 x i16> poison, i16 %205, i64 0, !dbg !32
+  %504 = insertelement <2 x i16> %503, i16 %453, i64 1, !dbg !32
+  %505 = bitcast <2 x i16> %504 to <2 x bfloat>, !dbg !32
+  %506 = fpext <2 x bfloat> %505 to <2 x float>, !dbg !58
+  %507 = insertelement <2 x float> poison, float %391, i64 0, !dbg !59
+  %508 = insertelement <2 x float> %507, float %347, i64 1, !dbg !59
+  %509 = fmul <2 x float> %508, %506, !dbg !59
+  %shift = shufflevector <2 x float> %460, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop = fadd <2 x float> %460, %shift, !dbg !60
+  %510 = extractelement <2 x float> %foldExtExtBinop, i64 0, !dbg !60
+  %shift66 = shufflevector <2 x float> %467, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop67 = fadd <2 x float> %467, %shift66, !dbg !60
+  %511 = extractelement <2 x float> %foldExtExtBinop67, i64 0, !dbg !60
+  %shift69 = shufflevector <2 x float> %474, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop70 = fadd <2 x float> %474, %shift69, !dbg !60
+  %512 = extractelement <2 x float> %foldExtExtBinop70, i64 0, !dbg !60
+  %shift72 = shufflevector <2 x float> %481, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop73 = fadd <2 x float> %481, %shift72, !dbg !60
+  %513 = extractelement <2 x float> %foldExtExtBinop73, i64 0, !dbg !60
+  %shift75 = shufflevector <2 x float> %488, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop76 = fadd <2 x float> %488, %shift75, !dbg !60
+  %514 = extractelement <2 x float> %foldExtExtBinop76, i64 0, !dbg !60
+  %shift78 = shufflevector <2 x float> %495, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop79 = fadd <2 x float> %495, %shift78, !dbg !60
+  %515 = extractelement <2 x float> %foldExtExtBinop79, i64 0, !dbg !60
+  %shift81 = shufflevector <2 x float> %502, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop82 = fadd <2 x float> %502, %shift81, !dbg !60
+  %516 = extractelement <2 x float> %foldExtExtBinop82, i64 0, !dbg !60
+  %shift84 = shufflevector <2 x float> %509, <2 x float> poison, <2 x i32> <i32 1, i32 poison>, !dbg !60
+  %foldExtExtBinop85 = fadd <2 x float> %509, %shift84, !dbg !60
+  %517 = extractelement <2 x float> %foldExtExtBinop85, i64 0, !dbg !60
+  %518 = getelementptr bfloat, ptr addrspace(1) %4, i64 %36, !dbg !61
+  %519 = getelementptr bfloat, ptr addrspace(1) %4, i64 %38, !dbg !61
+  %520 = getelementptr bfloat, ptr addrspace(1) %4, i64 %40, !dbg !61
+  %521 = getelementptr bfloat, ptr addrspace(1) %4, i64 %42, !dbg !61
+  %522 = getelementptr bfloat, ptr addrspace(1) %4, i64 %44, !dbg !61
+  %523 = getelementptr bfloat, ptr addrspace(1) %4, i64 %46, !dbg !61
+  %524 = getelementptr bfloat, ptr addrspace(1) %4, i64 %48, !dbg !61
+  %525 = getelementptr bfloat, ptr addrspace(1) %4, i64 %50, !dbg !61
+  %526 = fptrunc float %510 to bfloat, !dbg !62
+  %527 = fptrunc float %511 to bfloat, !dbg !62
+  %528 = fptrunc float %512 to bfloat, !dbg !62
+  %529 = fptrunc float %513 to bfloat, !dbg !62
+  %530 = fptrunc float %514 to bfloat, !dbg !62
+  %531 = fptrunc float %515 to bfloat, !dbg !62
+  %532 = fptrunc float %516 to bfloat, !dbg !62
+  %533 = fptrunc float %517 to bfloat, !dbg !62
+  %534 = bitcast bfloat %526 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %534, ptr addrspace(1) %518, i1 %88) #4, !dbg !62
+  %535 = bitcast bfloat %527 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %535, ptr addrspace(1) %519, i1 %91) #4, !dbg !62
+  %536 = bitcast bfloat %528 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %536, ptr addrspace(1) %520, i1 %94) #4, !dbg !62
+  %537 = bitcast bfloat %529 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %537, ptr addrspace(1) %521, i1 %97) #4, !dbg !62
+  %538 = bitcast bfloat %530 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %538, ptr addrspace(1) %522, i1 %100) #4, !dbg !62
+  %539 = bitcast bfloat %531 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %539, ptr addrspace(1) %523, i1 %103) #4, !dbg !62
+  %540 = bitcast bfloat %532 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %540, ptr addrspace(1) %524, i1 %106) #4, !dbg !62
+  %541 = bitcast bfloat %533 to i16, !dbg !62
+  tail call void asm sideeffect "@$2 st.global.b16 [ $1 + 0 ], { $0 };", "c,l,b"(i16 %541, ptr addrspace(1) %525, i1 %109) #4, !dbg !62
+  ret void, !dbg !63
+}
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 2147483647) i32 @llvm.nvvm.read.ptx.sreg.ctaid.x() #2
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 1024) i32 @llvm.nvvm.read.ptx.sreg.tid.x() #2
+
+; Function Attrs: convergent nocallback nounwind
+declare void @llvm.nvvm.barrier.cta.sync.aligned.all(i32) #3
+
+attributes #0 = { noreturn }
+attributes #1 = { "nvvm.reqntid"="128" }
+attributes #2 = { mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #3 = { convergent nocallback nounwind }
+attributes #4 = { nounwind }
+
+!llvm.dbg.cu = !{!0}
+!llvm.module.flags = !{!2, !3}
+!llvm.ident = !{!4}
+
+!0 = distinct !DICompileUnit(language: DW_LANG_C, file: !1, producer: "triton", isOptimized: true, runtimeVersion: 0, emissionKind: LineTablesOnly)
+!1 = !DIFile(filename: "cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py", directory: "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al")
+!2 = !{i32 2, !"Debug Info Version", i32 3}
+!3 = !{i32 4, !"nvvm-reflect-ftz", i32 1}
+!4 = !{!"clang version 3.8.0 (tags/RELEASE_380/final)"}
+!5 = !DISubprogram(name: "__assertfail", linkageName: "__assertfail", scope: !6, file: !6, type: !7, spFlags: DISPFlagOptimized)
+!6 = !DIFile(filename: "<unknown>", directory: "")
+!7 = !DISubroutineType(cc: DW_CC_normal, types: !8)
+!8 = !{}
+!9 = distinct !DISubprogram(name: "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0", linkageName: "triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0", scope: !1, file: !1, line: 18, type: !7, scopeLine: 18, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0)
+!10 = !DILocation(line: 19, column: 28, scope: !9)
+!11 = !DILocation(line: 19, column: 33, scope: !9)
+!12 = !DILocation(line: 20, column: 36, scope: !9)
+!13 = !DILocation(line: 20, column: 23, scope: !9)
+!14 = !DILocation(line: 23, column: 21, scope: !9)
+!15 = !DILocation(line: 26, column: 30, scope: !9)
+!16 = !DILocation(line: 23, column: 28, scope: !9)
+!17 = !DILocation(line: 26, column: 35, scope: !9)
+!18 = !DILocation(line: 27, column: 30, scope: !9)
+!19 = !DILocation(line: 21, column: 21, scope: !9)
+!20 = !DILocation(line: 27, column: 35, scope: !9)
+!21 = !DILocation(line: 30, column: 18, scope: !9)
+!22 = !DILocation(line: 31, column: 32, scope: !9)
+!23 = !DILocation(line: 32, column: 28, scope: !9)
+!24 = !DILocation(line: 32, column: 44, scope: !9)
+!25 = !DILocation(line: 32, column: 37, scope: !9)
+!26 = !DILocation(line: 32, column: 52, scope: !9)
+!27 = !DILocation(line: 32, column: 62, scope: !9)
+!28 = !DILocation(line: 24, column: 19, scope: !9)
+!29 = !DILocation(line: 33, column: 39, scope: !9)
+!30 = !DILocation(line: 40, column: 35, scope: !9)
+!31 = !DILocation(line: 33, column: 30, scope: !9)
+!32 = !DILocation(line: 33, column: 46, scope: !9)
+!33 = !DILocation(line: 38, column: 31, scope: !9)
+!34 = !DILocation(line: 38, column: 18, scope: !9)
+!35 = !DILocation(line: 39, column: 19, scope: !9)
+!36 = !DILocation(line: 40, column: 31, scope: !9)
+!37 = !DILocation(line: 40, column: 68, scope: !9)
+!38 = !DILocation(line: 40, column: 60, scope: !9)
+!39 = !DILocation(line: 44, column: 20, scope: !9)
+!40 = !DILocation(line: 47, column: 47, scope: !9)
+!41 = !DILocation(line: 47, column: 31, scope: !9)
+!42 = !DILocation(line: 47, column: 81, scope: !9)
+!43 = !DILocation(line: 47, column: 73, scope: !9)
+!44 = !DILocation(line: 51, column: 34, scope: !9)
+!45 = !DILocation(line: 52, column: 28, scope: !9)
+!46 = !DILocation(line: 52, column: 46, scope: !9)
+!47 = !DILocation(line: 52, column: 38, scope: !9)
+!48 = !DILocation(line: 52, column: 54, scope: !9)
+!49 = !DILocation(line: 52, column: 64, scope: !9)
+!50 = !DILocation(line: 40, column: 119, scope: !9)
+!51 = !DILocation(line: 41, column: 13, scope: !9)
+!52 = !DILocation(line: 47, column: 132, scope: !9)
+!53 = !DILocation(line: 0, scope: !9)
+!54 = !DILocation(line: 26, column: 75, scope: !9)
+!55 = !DILocation(line: 53, column: 40, scope: !9)
+!56 = !DILocation(line: 53, column: 31, scope: !9)
+!57 = !DILocation(line: 53, column: 48, scope: !9)
+!58 = !DILocation(line: 33, column: 86, scope: !9)
+!59 = !DILocation(line: 34, column: 18, scope: !9)
+!60 = !DILocation(line: 55, column: 19, scope: !9)
+!61 = !DILocation(line: 56, column: 25, scope: !9)
+!62 = !DILocation(line: 56, column: 37, scope: !9)
+!63 = !DILocation(line: 56, column: 4, scope: !9)

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ptx

@@ -0,0 +1,1534 @@
+//
+// Generated by LLVM NVPTX Back-End
+//
+
+.version 8.7
+.target sm_90a
+.address_size 64
+
+	// .globl	triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0 // -- Begin function triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0
+.extern .func __assertfail
+(
+	.param .b64 __assertfail_param_0,
+	.param .b64 __assertfail_param_1,
+	.param .b32 __assertfail_param_2,
+	.param .b64 __assertfail_param_3,
+	.param .b64 __assertfail_param_4
+)
+.noreturn;
+.global .align 1 .b8 assertFunc_1[8] = {117, 110, 107, 110, 111, 119, 110};
+.global .align 1 .b8 assertFile_1[114] = {47, 119, 111, 114, 107, 115, 112, 97, 99, 101, 47, 104, 97, 110, 114, 117, 105, 47, 83, 112, 101, 99, 70, 111, 114, 103, 101, 45, 101, 120, 116, 47, 99, 97, 99, 104, 101, 47, 99, 111, 109, 112, 105, 108, 101, 100, 95, 107, 101, 114, 110, 101, 108, 115, 47, 97, 108, 47, 99, 97, 108, 50, 114, 52, 116, 102, 121, 119, 54, 103, 105, 99, 51, 103, 103, 113, 121, 117, 100, 51, 110, 117, 102, 110, 97, 106, 120, 54, 120, 97, 117, 50, 107, 111, 105, 101, 111, 105, 116, 120, 54, 122, 103, 52, 119, 115, 105, 111, 122, 109, 46, 112, 121};
+.global .align 1 .b8 assertMessage_1[38] = {105, 110, 100, 101, 120, 32, 111, 117, 116, 32, 111, 102, 32, 98, 111, 117, 110, 100, 115, 58, 32, 48, 32, 60, 61, 32, 116, 109, 112, 50, 53, 32, 60, 32, 107, 115, 52};
+.global .align 1 .b8 assertFunc_0[8] = {117, 110, 107, 110, 111, 119, 110};
+.global .align 1 .b8 assertFile_0[114] = {47, 119, 111, 114, 107, 115, 112, 97, 99, 101, 47, 104, 97, 110, 114, 117, 105, 47, 83, 112, 101, 99, 70, 111, 114, 103, 101, 45, 101, 120, 116, 47, 99, 97, 99, 104, 101, 47, 99, 111, 109, 112, 105, 108, 101, 100, 95, 107, 101, 114, 110, 101, 108, 115, 47, 97, 108, 47, 99, 97, 108, 50, 114, 52, 116, 102, 121, 119, 54, 103, 105, 99, 51, 103, 103, 113, 121, 117, 100, 51, 110, 117, 102, 110, 97, 106, 120, 54, 120, 97, 117, 50, 107, 111, 105, 101, 111, 105, 116, 120, 54, 122, 103, 52, 119, 115, 105, 111, 122, 109, 46, 112, 121};
+.global .align 1 .b8 assertMessage_0[37] = {105, 110, 100, 101, 120, 32, 111, 117, 116, 32, 111, 102, 32, 98, 111, 117, 110, 100, 115, 58, 32, 48, 32, 60, 61, 32, 116, 109, 112, 53, 32, 60, 32, 107, 115, 50};
+                                        // @triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0
+.visible .entry triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0(
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_0,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_1,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_2,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_3,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_4,
+	.param .u64 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_5,
+	.param .u64 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_6,
+	.param .u64 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_7,
+	.param .u64 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_8,
+	.param .u64 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_9,
+	.param .u32 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_10,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_11,
+	.param .u64 .ptr .global .align 1 triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_12
+)
+.reqntid 128
+{
+	.reg .pred 	%p<179>;
+	.reg .b16 	%rs<165>;
+	.reg .b32 	%r<160>;
+	.reg .b64 	%rd<500>;
+	.loc	1 18 0                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:18:0
+$L__func_begin0:
+	.loc	1 18 0                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:18:0
+
+// %bb.0:
+	ld.param.b64 	%rd103, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_5];
+$L__tmp0:
+	.loc	1 19 28                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:19:28
+	mov.u32 	%r26, %ctaid.x;
+	.loc	1 19 33                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:19:33
+	shl.b32 	%r27, %r26, 10;
+	.loc	1 20 36                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:20:36
+	mov.u32 	%r28, %tid.x;
+	shl.b32 	%r29, %r28, 3;
+	and.b32 	%r30, %r29, 1016;
+	.loc	1 20 23                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:20:23
+	or.b32 	%r1, %r30, %r27;
+	or.b32 	%r2, %r1, 1;
+	or.b32 	%r3, %r1, 2;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd7, %r2;
+	cvt.s64.s32 	%rd8, %r1;
+	or.b64 	%rd108, %rd8, %rd103;
+	and.b64 	%rd109, %rd108, -4294967296;
+	setp.ne.b64 	%p9, %rd109, 0;
+	@%p9 bra 	$L__BB0_2;
+	bra.uni 	$L__BB0_1;
+$L__BB0_2:
+	div.s64 	%rd484, %rd8, %rd103;
+	bra.uni 	$L__BB0_3;
+$L__BB0_1:
+	cvt.u32.u64 	%r31, %rd103;
+	cvt.u32.u64 	%r32, %rd8;
+	div.u32 	%r33, %r32, %r31;
+	cvt.u64.u32 	%rd484, %r33;
+$L__BB0_3:
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	or.b32 	%r4, %r1, 3;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd6, %r3;
+	or.b64 	%rd110, %rd7, %rd103;
+	and.b64 	%rd111, %rd110, -4294967296;
+	setp.ne.b64 	%p10, %rd111, 0;
+	@%p10 bra 	$L__BB0_5;
+	bra.uni 	$L__BB0_4;
+$L__BB0_5:
+	div.s64 	%rd485, %rd7, %rd103;
+	bra.uni 	$L__BB0_6;
+$L__BB0_4:
+	cvt.u32.u64 	%r34, %rd103;
+	cvt.u32.u64 	%r35, %rd7;
+	div.u32 	%r36, %r35, %r34;
+	cvt.u64.u32 	%rd485, %r36;
+$L__BB0_6:
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	or.b32 	%r5, %r1, 4;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd5, %r4;
+	or.b64 	%rd112, %rd6, %rd103;
+	and.b64 	%rd113, %rd112, -4294967296;
+	setp.ne.b64 	%p11, %rd113, 0;
+	@%p11 bra 	$L__BB0_8;
+	bra.uni 	$L__BB0_7;
+$L__BB0_8:
+	div.s64 	%rd486, %rd6, %rd103;
+	bra.uni 	$L__BB0_9;
+$L__BB0_7:
+	cvt.u32.u64 	%r37, %rd103;
+	cvt.u32.u64 	%r38, %rd6;
+	div.u32 	%r39, %r38, %r37;
+	cvt.u64.u32 	%rd486, %r39;
+$L__BB0_9:
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	or.b32 	%r6, %r1, 5;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd4, %r5;
+	or.b64 	%rd114, %rd5, %rd103;
+	and.b64 	%rd115, %rd114, -4294967296;
+	setp.ne.b64 	%p12, %rd115, 0;
+	@%p12 bra 	$L__BB0_11;
+	bra.uni 	$L__BB0_10;
+$L__BB0_11:
+	div.s64 	%rd487, %rd5, %rd103;
+	bra.uni 	$L__BB0_12;
+$L__BB0_10:
+	cvt.u32.u64 	%r40, %rd103;
+	cvt.u32.u64 	%r41, %rd5;
+	div.u32 	%r42, %r41, %r40;
+	cvt.u64.u32 	%rd487, %r42;
+$L__BB0_12:
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	or.b32 	%r7, %r1, 6;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd3, %r6;
+	or.b64 	%rd116, %rd4, %rd103;
+	and.b64 	%rd117, %rd116, -4294967296;
+	setp.ne.b64 	%p13, %rd117, 0;
+	@%p13 bra 	$L__BB0_14;
+	bra.uni 	$L__BB0_13;
+$L__BB0_14:
+	div.s64 	%rd488, %rd4, %rd103;
+	bra.uni 	$L__BB0_15;
+$L__BB0_13:
+	cvt.u32.u64 	%r43, %rd103;
+	cvt.u32.u64 	%r44, %rd4;
+	div.u32 	%r45, %r44, %r43;
+	cvt.u64.u32 	%rd488, %r45;
+$L__BB0_15:
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	or.b32 	%r8, %r1, 7;
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	cvt.s64.s32 	%rd2, %r7;
+	or.b64 	%rd118, %rd3, %rd103;
+	and.b64 	%rd119, %rd118, -4294967296;
+	setp.ne.b64 	%p14, %rd119, 0;
+	@%p14 bra 	$L__BB0_17;
+	bra.uni 	$L__BB0_16;
+$L__BB0_17:
+	div.s64 	%rd489, %rd3, %rd103;
+	bra.uni 	$L__BB0_18;
+$L__BB0_16:
+	cvt.u32.u64 	%r46, %rd103;
+	cvt.u32.u64 	%r47, %rd3;
+	div.u32 	%r48, %r47, %r46;
+	cvt.u64.u32 	%rd489, %r48;
+$L__BB0_18:
+	cvt.s64.s32 	%rd1, %r8;
+	or.b64 	%rd120, %rd2, %rd103;
+	and.b64 	%rd121, %rd120, -4294967296;
+	setp.ne.b64 	%p15, %rd121, 0;
+	@%p15 bra 	$L__BB0_20;
+	bra.uni 	$L__BB0_19;
+$L__BB0_20:
+	div.s64 	%rd490, %rd2, %rd103;
+	bra.uni 	$L__BB0_21;
+$L__BB0_19:
+	cvt.u32.u64 	%r49, %rd103;
+	cvt.u32.u64 	%r50, %rd2;
+	div.u32 	%r51, %r50, %r49;
+	cvt.u64.u32 	%rd490, %r51;
+$L__BB0_21:
+	.loc	1 0 21                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0:21
+	ld.param.b64 	%rd104, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_6];
+	.loc	1 23 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:21
+	or.b64 	%rd122, %rd1, %rd103;
+	and.b64 	%rd123, %rd122, -4294967296;
+	setp.ne.b64 	%p16, %rd123, 0;
+	@%p16 bra 	$L__BB0_23;
+	bra.uni 	$L__BB0_22;
+$L__BB0_23:
+	div.s64 	%rd491, %rd1, %rd103;
+	bra.uni 	$L__BB0_24;
+$L__BB0_22:
+	cvt.u32.u64 	%r52, %rd103;
+	cvt.u32.u64 	%r53, %rd1;
+	div.u32 	%r54, %r53, %r52;
+	cvt.u64.u32 	%rd491, %r54;
+$L__BB0_24:
+	.loc	1 23 28                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:28
+	or.b64 	%rd124, %rd484, %rd104;
+	and.b64 	%rd125, %rd124, -4294967296;
+	setp.ne.b64 	%p17, %rd125, 0;
+	@%p17 bra 	$L__BB0_26;
+	bra.uni 	$L__BB0_25;
+$L__BB0_26:
+	rem.s64 	%rd492, %rd484, %rd104;
+	bra.uni 	$L__BB0_27;
+$L__BB0_25:
+	cvt.u32.u64 	%r55, %rd104;
+	cvt.u32.u64 	%r56, %rd484;
+	rem.u32 	%r57, %r56, %r55;
+	cvt.u64.u32 	%rd492, %r57;
+$L__BB0_27:
+	or.b64 	%rd126, %rd485, %rd104;
+	and.b64 	%rd127, %rd126, -4294967296;
+	setp.ne.b64 	%p18, %rd127, 0;
+	@%p18 bra 	$L__BB0_29;
+	bra.uni 	$L__BB0_28;
+$L__BB0_29:
+	rem.s64 	%rd493, %rd485, %rd104;
+	bra.uni 	$L__BB0_30;
+$L__BB0_28:
+	cvt.u32.u64 	%r58, %rd104;
+	cvt.u32.u64 	%r59, %rd485;
+	rem.u32 	%r60, %r59, %r58;
+	cvt.u64.u32 	%rd493, %r60;
+$L__BB0_30:
+	or.b64 	%rd128, %rd486, %rd104;
+	and.b64 	%rd129, %rd128, -4294967296;
+	setp.ne.b64 	%p19, %rd129, 0;
+	@%p19 bra 	$L__BB0_32;
+	bra.uni 	$L__BB0_31;
+$L__BB0_32:
+	rem.s64 	%rd494, %rd486, %rd104;
+	bra.uni 	$L__BB0_33;
+$L__BB0_31:
+	cvt.u32.u64 	%r61, %rd104;
+	cvt.u32.u64 	%r62, %rd486;
+	rem.u32 	%r63, %r62, %r61;
+	cvt.u64.u32 	%rd494, %r63;
+$L__BB0_33:
+	or.b64 	%rd130, %rd487, %rd104;
+	and.b64 	%rd131, %rd130, -4294967296;
+	setp.ne.b64 	%p20, %rd131, 0;
+	@%p20 bra 	$L__BB0_35;
+	bra.uni 	$L__BB0_34;
+$L__BB0_35:
+	rem.s64 	%rd495, %rd487, %rd104;
+	bra.uni 	$L__BB0_36;
+$L__BB0_34:
+	cvt.u32.u64 	%r64, %rd104;
+	cvt.u32.u64 	%r65, %rd487;
+	rem.u32 	%r66, %r65, %r64;
+	cvt.u64.u32 	%rd495, %r66;
+$L__BB0_36:
+	or.b64 	%rd132, %rd488, %rd104;
+	and.b64 	%rd133, %rd132, -4294967296;
+	setp.ne.b64 	%p21, %rd133, 0;
+	@%p21 bra 	$L__BB0_38;
+	bra.uni 	$L__BB0_37;
+$L__BB0_38:
+	rem.s64 	%rd496, %rd488, %rd104;
+	bra.uni 	$L__BB0_39;
+$L__BB0_37:
+	cvt.u32.u64 	%r67, %rd104;
+	cvt.u32.u64 	%r68, %rd488;
+	rem.u32 	%r69, %r68, %r67;
+	cvt.u64.u32 	%rd496, %r69;
+$L__BB0_39:
+	or.b64 	%rd134, %rd489, %rd104;
+	and.b64 	%rd135, %rd134, -4294967296;
+	setp.ne.b64 	%p22, %rd135, 0;
+	@%p22 bra 	$L__BB0_41;
+	bra.uni 	$L__BB0_40;
+$L__BB0_41:
+	rem.s64 	%rd497, %rd489, %rd104;
+	bra.uni 	$L__BB0_42;
+$L__BB0_40:
+	cvt.u32.u64 	%r70, %rd104;
+	cvt.u32.u64 	%r71, %rd489;
+	rem.u32 	%r72, %r71, %r70;
+	cvt.u64.u32 	%rd497, %r72;
+$L__BB0_42:
+	or.b64 	%rd136, %rd490, %rd104;
+	and.b64 	%rd137, %rd136, -4294967296;
+	setp.ne.b64 	%p23, %rd137, 0;
+	@%p23 bra 	$L__BB0_44;
+	bra.uni 	$L__BB0_43;
+$L__BB0_44:
+	rem.s64 	%rd498, %rd490, %rd104;
+	bra.uni 	$L__BB0_45;
+$L__BB0_43:
+	cvt.u32.u64 	%r73, %rd104;
+	cvt.u32.u64 	%r74, %rd490;
+	rem.u32 	%r75, %r74, %r73;
+	cvt.u64.u32 	%rd498, %r75;
+$L__BB0_45:
+	.loc	1 0 28                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0:28
+	ld.param.b32 	%r25, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_10];
+	ld.param.b64 	%rd105, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_7];
+	ld.param.b64 	%rd99, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_1];
+	ld.param.b64 	%rd98, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_0];
+	.loc	1 23 28                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:23:28
+	or.b64 	%rd138, %rd491, %rd104;
+	and.b64 	%rd139, %rd138, -4294967296;
+	setp.ne.b64 	%p24, %rd139, 0;
+	@%p24 bra 	$L__BB0_47;
+	bra.uni 	$L__BB0_46;
+$L__BB0_47:
+	rem.s64 	%rd499, %rd491, %rd104;
+	bra.uni 	$L__BB0_48;
+$L__BB0_46:
+	cvt.u32.u64 	%r76, %rd104;
+	cvt.u32.u64 	%r77, %rd491;
+	rem.u32 	%r78, %r77, %r76;
+	cvt.u64.u32 	%rd499, %r78;
+$L__BB0_48:
+	.loc	1 26 30                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:26:30
+	shl.b64 	%rd196, %rd8, 1;
+	add.s64 	%rd141, %rd98, %rd196;
+	shl.b64 	%rd197, %rd7, 1;
+	add.s64 	%rd144, %rd98, %rd197;
+	shl.b64 	%rd198, %rd6, 1;
+	add.s64 	%rd147, %rd98, %rd198;
+	shl.b64 	%rd199, %rd5, 1;
+	add.s64 	%rd150, %rd98, %rd199;
+	shl.b64 	%rd200, %rd4, 1;
+	add.s64 	%rd153, %rd98, %rd200;
+	shl.b64 	%rd201, %rd3, 1;
+	add.s64 	%rd156, %rd98, %rd201;
+	shl.b64 	%rd202, %rd2, 1;
+	add.s64 	%rd159, %rd98, %rd202;
+	shl.b64 	%rd203, %rd1, 1;
+	add.s64 	%rd162, %rd98, %rd203;
+	.loc	1 26 35                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:26:35
+	// begin inline asm
+	mov.u64 %rd140, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd140, 1.0;
+	// end inline asm
+	.loc	1 27 30                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:27:30
+	shl.b64 	%rd204, %rd492, 3;
+	add.s64 	%rd166, %rd99, %rd204;
+	shl.b64 	%rd205, %rd493, 3;
+	add.s64 	%rd170, %rd99, %rd205;
+	shl.b64 	%rd206, %rd494, 3;
+	add.s64 	%rd174, %rd99, %rd206;
+	shl.b64 	%rd207, %rd495, 3;
+	add.s64 	%rd178, %rd99, %rd207;
+	shl.b64 	%rd208, %rd496, 3;
+	add.s64 	%rd182, %rd99, %rd208;
+	shl.b64 	%rd209, %rd497, 3;
+	add.s64 	%rd186, %rd99, %rd209;
+	shl.b64 	%rd210, %rd498, 3;
+	add.s64 	%rd190, %rd99, %rd210;
+	shl.b64 	%rd211, %rd499, 3;
+	add.s64 	%rd194, %rd99, %rd211;
+	.loc	1 21 21                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:21:21
+	setp.lt.s32 	%p8, %r8, %r25;
+	setp.lt.s32 	%p7, %r7, %r25;
+	setp.lt.s32 	%p6, %r6, %r25;
+	setp.lt.s32 	%p5, %r5, %r25;
+	setp.lt.s32 	%p4, %r4, %r25;
+	setp.lt.s32 	%p3, %r3, %r25;
+	setp.lt.s32 	%p2, %r2, %r25;
+	setp.lt.s32 	%p1, %r1, %r25;
+	.loc	1 26 35                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:26:35
+	// begin inline asm
+	mov.u16 %rs33, 0x0;
+	@%p1 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs33 }, [ %rd141 + 0 ], %rd140;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd143, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd143, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs34, 0x0;
+	@%p2 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs34 }, [ %rd144 + 0 ], %rd143;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd146, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd146, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs35, 0x0;
+	@%p3 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs35 }, [ %rd147 + 0 ], %rd146;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd149, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd149, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs36, 0x0;
+	@%p4 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs36 }, [ %rd150 + 0 ], %rd149;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd152, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd152, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs37, 0x0;
+	@%p5 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs37 }, [ %rd153 + 0 ], %rd152;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd155, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd155, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs38, 0x0;
+	@%p6 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs38 }, [ %rd156 + 0 ], %rd155;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd158, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd158, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs39, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs39 }, [ %rd159 + 0 ], %rd158;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd161, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd161, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs40, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs40 }, [ %rd162 + 0 ], %rd161;
+	// end inline asm
+	.loc	1 27 35                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:27:35
+	// begin inline asm
+	mov.u64 %rd164, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd164, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd165, 0x0;
+	@%p1 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd165 }, [ %rd166 + 0 ], %rd164;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd168, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd168, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd169, 0x0;
+	@%p2 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd169 }, [ %rd170 + 0 ], %rd168;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd172, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd172, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd173, 0x0;
+	@%p3 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd173 }, [ %rd174 + 0 ], %rd172;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd176, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd176, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd177, 0x0;
+	@%p4 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd177 }, [ %rd178 + 0 ], %rd176;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd180, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd180, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd181, 0x0;
+	@%p5 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd181 }, [ %rd182 + 0 ], %rd180;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd184, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd184, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd185, 0x0;
+	@%p6 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd185 }, [ %rd186 + 0 ], %rd184;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd188, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd188, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd189, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd189 }, [ %rd190 + 0 ], %rd188;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd192, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd192, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd193, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b64 { %rd193 }, [ %rd194 + 0 ], %rd192;
+	// end inline asm
+	.loc	1 31 32                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:31:32
+	shr.s64 	%rd212, %rd173, 63;
+	and.b64 	%rd213, %rd212, %rd105;
+	shr.s64 	%rd214, %rd177, 63;
+	and.b64 	%rd215, %rd214, %rd105;
+	shr.s64 	%rd216, %rd165, 63;
+	and.b64 	%rd217, %rd216, %rd105;
+	shr.s64 	%rd218, %rd169, 63;
+	and.b64 	%rd219, %rd218, %rd105;
+	shr.s64 	%rd220, %rd189, 63;
+	and.b64 	%rd221, %rd220, %rd105;
+	shr.s64 	%rd222, %rd193, 63;
+	and.b64 	%rd223, %rd222, %rd105;
+	shr.s64 	%rd224, %rd181, 63;
+	and.b64 	%rd225, %rd224, %rd105;
+	shr.s64 	%rd226, %rd185, 63;
+	and.b64 	%rd227, %rd226, %rd105;
+	add.s64 	%rd78, %rd227, %rd185;
+	add.s64 	%rd77, %rd225, %rd181;
+	add.s64 	%rd80, %rd223, %rd193;
+	add.s64 	%rd79, %rd221, %rd189;
+	add.s64 	%rd74, %rd219, %rd169;
+	add.s64 	%rd73, %rd217, %rd165;
+	add.s64 	%rd76, %rd215, %rd177;
+	add.s64 	%rd75, %rd213, %rd173;
+	.loc	1 32 28                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:28
+	setp.lt.s64 	%p41, %rd75, 0;
+	setp.lt.s64 	%p42, %rd76, 0;
+	setp.lt.s64 	%p43, %rd73, 0;
+	setp.lt.s64 	%p44, %rd74, 0;
+	setp.lt.s64 	%p45, %rd79, 0;
+	setp.lt.s64 	%p46, %rd80, 0;
+	setp.lt.s64 	%p47, %rd77, 0;
+	setp.lt.s64 	%p48, %rd78, 0;
+	.loc	1 32 44                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:44
+	setp.ge.s64 	%p49, %rd75, %rd105;
+	setp.ge.s64 	%p50, %rd76, %rd105;
+	setp.ge.s64 	%p51, %rd73, %rd105;
+	setp.ge.s64 	%p52, %rd74, %rd105;
+	setp.ge.s64 	%p53, %rd79, %rd105;
+	setp.ge.s64 	%p54, %rd80, %rd105;
+	setp.ge.s64 	%p55, %rd77, %rd105;
+	setp.ge.s64 	%p56, %rd78, %rd105;
+	.loc	1 32 37                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:37
+	or.pred 	%p57, %p48, %p56;
+	or.pred 	%p58, %p47, %p55;
+	or.pred 	%p59, %p46, %p54;
+	or.pred 	%p60, %p45, %p53;
+	or.pred 	%p61, %p44, %p52;
+	or.pred 	%p62, %p43, %p51;
+	or.pred 	%p63, %p42, %p50;
+	or.pred 	%p64, %p41, %p49;
+	.loc	1 32 52                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:52
+	and.pred 	%p65, %p3, %p64;
+	selp.b16 	%rs41, 1, 0, %p65;
+	shl.b16 	%rs42, %rs41, 2;
+	and.pred 	%p66, %p4, %p63;
+	selp.b16 	%rs43, -1, 0, %p66;
+	shl.b16 	%rs44, %rs43, 3;
+	or.b16 	%rs45, %rs44, %rs42;
+	and.pred 	%p67, %p1, %p62;
+	selp.b16 	%rs46, 1, 0, %p67;
+	and.pred 	%p68, %p2, %p61;
+	selp.b16 	%rs47, -1, 0, %p68;
+	shl.b16 	%rs48, %rs47, 1;
+	or.b16 	%rs49, %rs46, %rs48;
+	and.b16 	%rs50, %rs49, 3;
+	or.b16 	%rs51, %rs50, %rs45;
+	and.b16 	%rs52, %rs51, 15;
+	and.pred 	%p69, %p7, %p60;
+	selp.b16 	%rs53, 1, 0, %p69;
+	shl.b16 	%rs54, %rs53, 2;
+	and.pred 	%p70, %p8, %p59;
+	selp.b16 	%rs55, -1, 0, %p70;
+	shl.b16 	%rs56, %rs55, 3;
+	or.b16 	%rs57, %rs56, %rs54;
+	and.pred 	%p71, %p5, %p58;
+	selp.b16 	%rs58, 1, 0, %p71;
+	and.pred 	%p72, %p6, %p57;
+	selp.b16 	%rs59, -1, 0, %p72;
+	shl.b16 	%rs60, %rs59, 1;
+	or.b16 	%rs61, %rs58, %rs60;
+	and.b16 	%rs62, %rs61, 3;
+	or.b16 	%rs63, %rs62, %rs57;
+	shl.b16 	%rs64, %rs63, 4;
+	or.b16 	%rs65, %rs52, %rs64;
+	.loc	1 32 62                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:62
+	and.b16 	%rs66, %rs65, 255;
+	setp.eq.b16 	%p73, %rs66, 0;
+	@%p73 bra 	$L__BB0_50;
+	bra.uni 	$L__BB0_49;
+$L__BB0_50:
+	.loc	1 0 62                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0:62
+	ld.param.b64 	%rd107, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_9];
+	ld.param.b64 	%rd106, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_8];
+	ld.param.b64 	%rd100, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_2];
+	.loc	1 24 19                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:24:19
+	rem.s64 	%rd88, %rd1, %rd106;
+	rem.s64 	%rd87, %rd2, %rd106;
+	rem.s64 	%rd86, %rd3, %rd106;
+	rem.s64 	%rd85, %rd4, %rd106;
+	rem.s64 	%rd84, %rd5, %rd106;
+	rem.s64 	%rd83, %rd6, %rd106;
+	rem.s64 	%rd82, %rd7, %rd106;
+	rem.s64 	%rd81, %rd8, %rd106;
+	.loc	1 32 62                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:62
+	bar.sync 	0;
+	.loc	1 33 39                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:39
+	mul.lo.s64 	%rd306, %rd73, %rd106;
+	mul.lo.s64 	%rd307, %rd74, %rd106;
+	mul.lo.s64 	%rd308, %rd75, %rd106;
+	mul.lo.s64 	%rd309, %rd76, %rd106;
+	mul.lo.s64 	%rd310, %rd77, %rd106;
+	mul.lo.s64 	%rd311, %rd78, %rd106;
+	mul.lo.s64 	%rd312, %rd79, %rd106;
+	mul.lo.s64 	%rd313, %rd80, %rd106;
+	.loc	1 33 30                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:30
+	shl.b64 	%rd314, %rd81, 1;
+	add.s64 	%rd315, %rd100, %rd314;
+	shl.b64 	%rd316, %rd306, 1;
+	add.s64 	%rd235, %rd315, %rd316;
+	shl.b64 	%rd317, %rd82, 1;
+	add.s64 	%rd318, %rd100, %rd317;
+	shl.b64 	%rd319, %rd307, 1;
+	add.s64 	%rd238, %rd318, %rd319;
+	shl.b64 	%rd320, %rd83, 1;
+	add.s64 	%rd321, %rd100, %rd320;
+	shl.b64 	%rd322, %rd308, 1;
+	add.s64 	%rd241, %rd321, %rd322;
+	shl.b64 	%rd323, %rd84, 1;
+	add.s64 	%rd324, %rd100, %rd323;
+	shl.b64 	%rd325, %rd309, 1;
+	add.s64 	%rd244, %rd324, %rd325;
+	shl.b64 	%rd326, %rd85, 1;
+	add.s64 	%rd327, %rd100, %rd326;
+	shl.b64 	%rd328, %rd310, 1;
+	add.s64 	%rd247, %rd327, %rd328;
+	shl.b64 	%rd329, %rd86, 1;
+	add.s64 	%rd330, %rd100, %rd329;
+	shl.b64 	%rd331, %rd311, 1;
+	add.s64 	%rd250, %rd330, %rd331;
+	shl.b64 	%rd332, %rd87, 1;
+	add.s64 	%rd333, %rd100, %rd332;
+	shl.b64 	%rd334, %rd312, 1;
+	add.s64 	%rd253, %rd333, %rd334;
+	shl.b64 	%rd335, %rd88, 1;
+	add.s64 	%rd336, %rd100, %rd335;
+	shl.b64 	%rd337, %rd313, 1;
+	add.s64 	%rd256, %rd336, %rd337;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	// begin inline asm
+	mov.u64 %rd234, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd234, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs67, 0x0;
+	@%p1 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs67 }, [ %rd235 + 0 ], %rd234;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd237, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd237, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs68, 0x0;
+	@%p2 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs68 }, [ %rd238 + 0 ], %rd237;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd240, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd240, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs69, 0x0;
+	@%p3 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs69 }, [ %rd241 + 0 ], %rd240;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd243, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd243, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs70, 0x0;
+	@%p4 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs70 }, [ %rd244 + 0 ], %rd243;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd246, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd246, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs71, 0x0;
+	@%p5 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs71 }, [ %rd247 + 0 ], %rd246;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd249, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd249, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs72, 0x0;
+	@%p6 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs72 }, [ %rd250 + 0 ], %rd249;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd252, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd252, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs73, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs73 }, [ %rd253 + 0 ], %rd252;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd255, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd255, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs74, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs74 }, [ %rd256 + 0 ], %rd255;
+	// end inline asm
+	.loc	1 38 31                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:38:31
+	shr.u64 	%rd338, %rd106, 63;
+	add.s64 	%rd339, %rd106, %rd338;
+	shr.s64 	%rd340, %rd339, 1;
+	.loc	1 38 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:38:18
+	sub.s64 	%rd89, %rd106, %rd340;
+	.loc	1 39 19                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:39:19
+	setp.lt.s64 	%p106, %rd81, %rd89;
+	setp.lt.s64 	%p107, %rd82, %rd89;
+	setp.lt.s64 	%p108, %rd83, %rd89;
+	setp.lt.s64 	%p109, %rd84, %rd89;
+	setp.lt.s64 	%p110, %rd85, %rd89;
+	setp.lt.s64 	%p111, %rd86, %rd89;
+	setp.lt.s64 	%p112, %rd87, %rd89;
+	setp.lt.s64 	%p113, %rd88, %rd89;
+	.loc	1 40 35                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:35
+	sub.s64 	%rd341, %rd8, %rd81;
+	sub.s64 	%rd342, %rd7, %rd82;
+	sub.s64 	%rd343, %rd6, %rd83;
+	sub.s64 	%rd344, %rd5, %rd84;
+	sub.s64 	%rd345, %rd4, %rd85;
+	sub.s64 	%rd346, %rd3, %rd86;
+	sub.s64 	%rd347, %rd2, %rd87;
+	sub.s64 	%rd348, %rd1, %rd88;
+	.loc	1 40 31                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:31
+	shl.b64 	%rd349, %rd341, 1;
+	add.s64 	%rd350, %rd98, %rd349;
+	and.b64 	%rd351, %rd339, -2;
+	add.s64 	%rd352, %rd350, %rd351;
+	add.s64 	%rd259, %rd352, %rd314;
+	shl.b64 	%rd353, %rd342, 1;
+	add.s64 	%rd354, %rd98, %rd353;
+	add.s64 	%rd355, %rd354, %rd351;
+	add.s64 	%rd262, %rd355, %rd317;
+	shl.b64 	%rd356, %rd343, 1;
+	add.s64 	%rd357, %rd98, %rd356;
+	add.s64 	%rd358, %rd357, %rd351;
+	add.s64 	%rd265, %rd358, %rd320;
+	shl.b64 	%rd359, %rd344, 1;
+	add.s64 	%rd360, %rd98, %rd359;
+	add.s64 	%rd361, %rd360, %rd351;
+	add.s64 	%rd268, %rd361, %rd323;
+	shl.b64 	%rd362, %rd345, 1;
+	add.s64 	%rd363, %rd98, %rd362;
+	add.s64 	%rd364, %rd363, %rd351;
+	add.s64 	%rd271, %rd364, %rd326;
+	shl.b64 	%rd365, %rd346, 1;
+	add.s64 	%rd366, %rd98, %rd365;
+	add.s64 	%rd367, %rd366, %rd351;
+	add.s64 	%rd274, %rd367, %rd329;
+	shl.b64 	%rd368, %rd347, 1;
+	add.s64 	%rd369, %rd98, %rd368;
+	add.s64 	%rd370, %rd369, %rd351;
+	add.s64 	%rd277, %rd370, %rd332;
+	shl.b64 	%rd371, %rd348, 1;
+	add.s64 	%rd372, %rd98, %rd371;
+	add.s64 	%rd373, %rd372, %rd351;
+	add.s64 	%rd280, %rd373, %rd335;
+	.loc	1 40 68                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:68
+	and.pred 	%p82, %p1, %p106;
+	and.pred 	%p83, %p2, %p107;
+	and.pred 	%p84, %p3, %p108;
+	and.pred 	%p85, %p4, %p109;
+	and.pred 	%p86, %p5, %p110;
+	and.pred 	%p87, %p6, %p111;
+	and.pred 	%p88, %p7, %p112;
+	and.pred 	%p89, %p8, %p113;
+	.loc	1 40 60                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:60
+	// begin inline asm
+	mov.u64 %rd258, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd258, 1.0;
+	// end inline asm
+	mov.b16 	%rs76, 0;
+	// begin inline asm
+	mov.u16 %rs75, %rs76;
+	@%p82 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs75 }, [ %rd259 + 0 ], %rd258;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd261, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd261, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs77, %rs76;
+	@%p83 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs77 }, [ %rd262 + 0 ], %rd261;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd264, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd264, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs79, %rs76;
+	@%p84 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs79 }, [ %rd265 + 0 ], %rd264;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd267, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd267, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs81, %rs76;
+	@%p85 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs81 }, [ %rd268 + 0 ], %rd267;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd270, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd270, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs83, %rs76;
+	@%p86 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs83 }, [ %rd271 + 0 ], %rd270;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd273, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd273, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs85, %rs76;
+	@%p87 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs85 }, [ %rd274 + 0 ], %rd273;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd276, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd276, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs87, %rs76;
+	@%p88 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs87 }, [ %rd277 + 0 ], %rd276;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd279, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd279, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs89, %rs76;
+	@%p89 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs89 }, [ %rd280 + 0 ], %rd279;
+	// end inline asm
+	.loc	1 44 20                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:44:20
+	setp.ge.s64 	%p114, %rd88, %rd89;
+	setp.ge.s64 	%p115, %rd87, %rd89;
+	setp.ge.s64 	%p116, %rd86, %rd89;
+	setp.ge.s64 	%p117, %rd85, %rd89;
+	setp.ge.s64 	%p118, %rd84, %rd89;
+	setp.ge.s64 	%p119, %rd83, %rd89;
+	setp.ge.s64 	%p120, %rd82, %rd89;
+	setp.ge.s64 	%p121, %rd81, %rd89;
+	.loc	1 47 47                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:47
+	sub.s64 	%rd374, %rd81, %rd106;
+	sub.s64 	%rd375, %rd82, %rd106;
+	sub.s64 	%rd376, %rd83, %rd106;
+	sub.s64 	%rd377, %rd84, %rd106;
+	sub.s64 	%rd378, %rd85, %rd106;
+	sub.s64 	%rd379, %rd86, %rd106;
+	sub.s64 	%rd380, %rd87, %rd106;
+	sub.s64 	%rd381, %rd88, %rd106;
+	.loc	1 47 31                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:31
+	shl.b64 	%rd382, %rd374, 1;
+	add.s64 	%rd283, %rd352, %rd382;
+	shl.b64 	%rd383, %rd375, 1;
+	add.s64 	%rd286, %rd355, %rd383;
+	shl.b64 	%rd384, %rd376, 1;
+	add.s64 	%rd289, %rd358, %rd384;
+	shl.b64 	%rd385, %rd377, 1;
+	add.s64 	%rd292, %rd361, %rd385;
+	shl.b64 	%rd386, %rd378, 1;
+	add.s64 	%rd295, %rd364, %rd386;
+	shl.b64 	%rd387, %rd379, 1;
+	add.s64 	%rd298, %rd367, %rd387;
+	shl.b64 	%rd388, %rd380, 1;
+	add.s64 	%rd301, %rd370, %rd388;
+	shl.b64 	%rd389, %rd381, 1;
+	add.s64 	%rd304, %rd373, %rd389;
+	.loc	1 47 81                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:81
+	and.pred 	%p90, %p1, %p121;
+	and.pred 	%p91, %p2, %p120;
+	and.pred 	%p92, %p3, %p119;
+	and.pred 	%p93, %p4, %p118;
+	and.pred 	%p94, %p5, %p117;
+	and.pred 	%p95, %p6, %p116;
+	and.pred 	%p96, %p7, %p115;
+	and.pred 	%p97, %p8, %p114;
+	.loc	1 47 73                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:73
+	// begin inline asm
+	mov.u64 %rd282, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd282, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs91, %rs76;
+	@%p90 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs91 }, [ %rd283 + 0 ], %rd282;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd285, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd285, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs93, %rs76;
+	@%p91 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs93 }, [ %rd286 + 0 ], %rd285;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd288, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd288, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs95, %rs76;
+	@%p92 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs95 }, [ %rd289 + 0 ], %rd288;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd291, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd291, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs97, %rs76;
+	@%p93 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs97 }, [ %rd292 + 0 ], %rd291;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd294, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd294, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs99, %rs76;
+	@%p94 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs99 }, [ %rd295 + 0 ], %rd294;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd297, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd297, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs101, %rs76;
+	@%p95 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs101 }, [ %rd298 + 0 ], %rd297;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd300, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd300, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs103, %rs76;
+	@%p96 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs103 }, [ %rd301 + 0 ], %rd300;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd303, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd303, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs105, %rs76;
+	@%p97 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs105 }, [ %rd304 + 0 ], %rd303;
+	// end inline asm
+	.loc	1 51 34                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:51:34
+	and.b64 	%rd391, %rd212, %rd107;
+	and.b64 	%rd393, %rd214, %rd107;
+	and.b64 	%rd395, %rd216, %rd107;
+	and.b64 	%rd397, %rd218, %rd107;
+	and.b64 	%rd399, %rd220, %rd107;
+	and.b64 	%rd401, %rd222, %rd107;
+	and.b64 	%rd403, %rd224, %rd107;
+	and.b64 	%rd405, %rd226, %rd107;
+	add.s64 	%rd95, %rd405, %rd185;
+	add.s64 	%rd94, %rd403, %rd181;
+	add.s64 	%rd97, %rd401, %rd193;
+	add.s64 	%rd96, %rd399, %rd189;
+	add.s64 	%rd91, %rd397, %rd169;
+	add.s64 	%rd90, %rd395, %rd165;
+	add.s64 	%rd93, %rd393, %rd177;
+	add.s64 	%rd92, %rd391, %rd173;
+	.loc	1 52 28                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:28
+	setp.lt.s64 	%p122, %rd92, 0;
+	setp.lt.s64 	%p123, %rd93, 0;
+	setp.lt.s64 	%p124, %rd90, 0;
+	setp.lt.s64 	%p125, %rd91, 0;
+	setp.lt.s64 	%p126, %rd96, 0;
+	setp.lt.s64 	%p127, %rd97, 0;
+	setp.lt.s64 	%p128, %rd94, 0;
+	setp.lt.s64 	%p129, %rd95, 0;
+	.loc	1 52 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:46
+	setp.ge.s64 	%p130, %rd92, %rd107;
+	setp.ge.s64 	%p131, %rd93, %rd107;
+	setp.ge.s64 	%p132, %rd90, %rd107;
+	setp.ge.s64 	%p133, %rd91, %rd107;
+	setp.ge.s64 	%p134, %rd96, %rd107;
+	setp.ge.s64 	%p135, %rd97, %rd107;
+	setp.ge.s64 	%p136, %rd94, %rd107;
+	setp.ge.s64 	%p137, %rd95, %rd107;
+	.loc	1 52 38                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:38
+	or.pred 	%p138, %p129, %p137;
+	or.pred 	%p139, %p128, %p136;
+	or.pred 	%p140, %p127, %p135;
+	or.pred 	%p141, %p126, %p134;
+	or.pred 	%p142, %p125, %p133;
+	or.pred 	%p143, %p124, %p132;
+	or.pred 	%p144, %p123, %p131;
+	or.pred 	%p145, %p122, %p130;
+	.loc	1 52 54                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:54
+	and.pred 	%p146, %p3, %p145;
+	selp.b16 	%rs107, 1, 0, %p146;
+	shl.b16 	%rs108, %rs107, 2;
+	and.pred 	%p147, %p4, %p144;
+	selp.b16 	%rs109, -1, 0, %p147;
+	shl.b16 	%rs110, %rs109, 3;
+	or.b16 	%rs111, %rs110, %rs108;
+	and.pred 	%p148, %p1, %p143;
+	selp.b16 	%rs112, 1, 0, %p148;
+	and.pred 	%p149, %p2, %p142;
+	selp.b16 	%rs113, -1, 0, %p149;
+	shl.b16 	%rs114, %rs113, 1;
+	or.b16 	%rs115, %rs112, %rs114;
+	and.b16 	%rs116, %rs115, 3;
+	or.b16 	%rs117, %rs116, %rs111;
+	and.b16 	%rs118, %rs117, 15;
+	and.pred 	%p150, %p7, %p141;
+	selp.b16 	%rs119, 1, 0, %p150;
+	shl.b16 	%rs120, %rs119, 2;
+	and.pred 	%p151, %p8, %p140;
+	selp.b16 	%rs121, -1, 0, %p151;
+	shl.b16 	%rs122, %rs121, 3;
+	or.b16 	%rs123, %rs122, %rs120;
+	and.pred 	%p152, %p5, %p139;
+	selp.b16 	%rs124, 1, 0, %p152;
+	and.pred 	%p153, %p6, %p138;
+	selp.b16 	%rs125, -1, 0, %p153;
+	shl.b16 	%rs126, %rs125, 1;
+	or.b16 	%rs127, %rs124, %rs126;
+	and.b16 	%rs128, %rs127, 3;
+	or.b16 	%rs129, %rs128, %rs123;
+	shl.b16 	%rs130, %rs129, 4;
+	or.b16 	%rs131, %rs118, %rs130;
+	.loc	1 52 64                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:64
+	and.b16 	%rs132, %rs131, 255;
+	setp.eq.b16 	%p154, %rs132, 0;
+	@%p154 bra 	$L__BB0_52;
+	bra.uni 	$L__BB0_51;
+$L__BB0_52:
+	.loc	1 0 64                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0:64
+	ld.param.b64 	%rd102, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_4];
+	ld.param.b64 	%rd101, [triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0_param_3];
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r79, %rs89;
+	mov.b32 	%r80, 0f00000000;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r81, %r80, %r79;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r82, %rs105;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r83, %r81, %r82, %p113;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r84, %rs87;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r85, %r80, %r84;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r86, %rs103;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r87, %r85, %r86, %p112;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r88, %rs85;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r89, %r80, %r88;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r90, %rs101;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r91, %r89, %r90, %p111;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r92, %rs83;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r93, %r80, %r92;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r94, %rs99;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r95, %r93, %r94, %p110;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r96, %rs81;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r97, %r80, %r96;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r98, %rs97;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r99, %r97, %r98, %p109;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r100, %rs79;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r101, %r80, %r100;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r102, %rs95;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r103, %r101, %r102, %p108;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r104, %rs77;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r105, %r80, %r104;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r106, %rs93;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r107, %r105, %r106, %p107;
+	.loc	1 40 119                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:40:119
+	cvt.f32.bf16 	%r108, %rs75;
+	.loc	1 41 13                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:41:13
+	sub.f32 	%r109, %r80, %r108;
+	.loc	1 47 132                        // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:47:132
+	cvt.f32.bf16 	%r110, %rs91;
+	.loc	1 0 0                           // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:0
+	selp.f32 	%r111, %r109, %r110, %p106;
+	.loc	1 26 75                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:26:75
+	cvt.f32.bf16 	%r112, %rs40;
+	cvt.f32.bf16 	%r113, %rs39;
+	cvt.f32.bf16 	%r114, %rs38;
+	cvt.f32.bf16 	%r115, %rs37;
+	cvt.f32.bf16 	%r116, %rs36;
+	cvt.f32.bf16 	%r117, %rs35;
+	cvt.f32.bf16 	%r118, %rs34;
+	cvt.f32.bf16 	%r119, %rs33;
+	.loc	1 52 64                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:64
+	bar.sync 	0;
+	.loc	1 53 40                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:53:40
+	mul.lo.s64 	%rd444, %rd90, %rd106;
+	mul.lo.s64 	%rd445, %rd91, %rd106;
+	mul.lo.s64 	%rd446, %rd92, %rd106;
+	mul.lo.s64 	%rd447, %rd93, %rd106;
+	mul.lo.s64 	%rd448, %rd94, %rd106;
+	mul.lo.s64 	%rd449, %rd95, %rd106;
+	mul.lo.s64 	%rd450, %rd96, %rd106;
+	mul.lo.s64 	%rd451, %rd97, %rd106;
+	.loc	1 53 31                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:53:31
+	add.s64 	%rd453, %rd101, %rd314;
+	shl.b64 	%rd454, %rd444, 1;
+	add.s64 	%rd413, %rd453, %rd454;
+	add.s64 	%rd456, %rd101, %rd317;
+	shl.b64 	%rd457, %rd445, 1;
+	add.s64 	%rd416, %rd456, %rd457;
+	add.s64 	%rd459, %rd101, %rd320;
+	shl.b64 	%rd460, %rd446, 1;
+	add.s64 	%rd419, %rd459, %rd460;
+	add.s64 	%rd462, %rd101, %rd323;
+	shl.b64 	%rd463, %rd447, 1;
+	add.s64 	%rd422, %rd462, %rd463;
+	add.s64 	%rd465, %rd101, %rd326;
+	shl.b64 	%rd466, %rd448, 1;
+	add.s64 	%rd425, %rd465, %rd466;
+	add.s64 	%rd468, %rd101, %rd329;
+	shl.b64 	%rd469, %rd449, 1;
+	add.s64 	%rd428, %rd468, %rd469;
+	add.s64 	%rd471, %rd101, %rd332;
+	shl.b64 	%rd472, %rd450, 1;
+	add.s64 	%rd431, %rd471, %rd472;
+	add.s64 	%rd474, %rd101, %rd335;
+	shl.b64 	%rd475, %rd451, 1;
+	add.s64 	%rd434, %rd474, %rd475;
+	.loc	1 53 48                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:53:48
+	// begin inline asm
+	mov.u64 %rd414, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd414, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs133, 0x0;
+	@%p1 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs133 }, [ %rd413 + 0 ], %rd414;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd417, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd417, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs134, 0x0;
+	@%p2 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs134 }, [ %rd416 + 0 ], %rd417;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd420, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd420, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs135, 0x0;
+	@%p3 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs135 }, [ %rd419 + 0 ], %rd420;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd423, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd423, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs136, 0x0;
+	@%p4 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs136 }, [ %rd422 + 0 ], %rd423;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd426, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd426, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs137, 0x0;
+	@%p5 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs137 }, [ %rd425 + 0 ], %rd426;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd429, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd429, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs138, 0x0;
+	@%p6 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs138 }, [ %rd428 + 0 ], %rd429;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd432, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd432, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs139, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs139 }, [ %rd431 + 0 ], %rd432;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd435, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd435, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u16 %rs140, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b16 { %rs140 }, [ %rd434 + 0 ], %rd435;
+	// end inline asm
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r120, {%rs67, %rs133};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs149, %rs150}, %r120;
+	cvt.f32.bf16 	%r121, %rs149;
+	cvt.f32.bf16 	%r122, %rs150;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r123, %r111, %r122;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r124, {%rs68, %rs134};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs151, %rs152}, %r124;
+	cvt.f32.bf16 	%r125, %rs151;
+	cvt.f32.bf16 	%r126, %rs152;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r127, %r107, %r126;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r128, {%rs69, %rs135};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs153, %rs154}, %r128;
+	cvt.f32.bf16 	%r129, %rs153;
+	cvt.f32.bf16 	%r130, %rs154;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r131, %r103, %r130;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r132, {%rs70, %rs136};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs155, %rs156}, %r132;
+	cvt.f32.bf16 	%r133, %rs155;
+	cvt.f32.bf16 	%r134, %rs156;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r135, %r99, %r134;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r136, {%rs71, %rs137};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs157, %rs158}, %r136;
+	cvt.f32.bf16 	%r137, %rs157;
+	cvt.f32.bf16 	%r138, %rs158;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r139, %r95, %r138;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r140, {%rs72, %rs138};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs159, %rs160}, %r140;
+	cvt.f32.bf16 	%r141, %rs159;
+	cvt.f32.bf16 	%r142, %rs160;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r143, %r91, %r142;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r144, {%rs73, %rs139};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs161, %rs162}, %r144;
+	cvt.f32.bf16 	%r145, %rs161;
+	cvt.f32.bf16 	%r146, %rs162;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r147, %r87, %r146;
+	.loc	1 33 46                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:46
+	mov.b32 	%r148, {%rs74, %rs140};
+	.loc	1 33 86                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:33:86
+	mov.b32 	{%rs163, %rs164}, %r148;
+	cvt.f32.bf16 	%r149, %rs163;
+	cvt.f32.bf16 	%r150, %rs164;
+	.loc	1 34 18                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:34:18
+	mul.f32 	%r151, %r83, %r150;
+	.loc	1 55 19                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:55:19
+	fma.rn.f32 	%r152, %r119, %r121, %r123;
+	fma.rn.f32 	%r153, %r118, %r125, %r127;
+	fma.rn.f32 	%r154, %r117, %r129, %r131;
+	fma.rn.f32 	%r155, %r116, %r133, %r135;
+	fma.rn.f32 	%r156, %r115, %r137, %r139;
+	fma.rn.f32 	%r157, %r114, %r141, %r143;
+	fma.rn.f32 	%r158, %r113, %r145, %r147;
+	fma.rn.f32 	%r159, %r112, %r149, %r151;
+	.loc	1 56 25                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:56:25
+	add.s64 	%rd436, %rd102, %rd196;
+	add.s64 	%rd437, %rd102, %rd197;
+	add.s64 	%rd438, %rd102, %rd198;
+	add.s64 	%rd439, %rd102, %rd199;
+	add.s64 	%rd440, %rd102, %rd200;
+	add.s64 	%rd441, %rd102, %rd201;
+	add.s64 	%rd442, %rd102, %rd202;
+	add.s64 	%rd443, %rd102, %rd203;
+	.loc	1 56 37                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:56:37
+	cvt.rn.bf16.f32 	%rs141, %r152;
+	cvt.rn.bf16.f32 	%rs142, %r153;
+	cvt.rn.bf16.f32 	%rs143, %r154;
+	cvt.rn.bf16.f32 	%rs144, %r155;
+	cvt.rn.bf16.f32 	%rs145, %r156;
+	cvt.rn.bf16.f32 	%rs146, %r157;
+	cvt.rn.bf16.f32 	%rs147, %r158;
+	cvt.rn.bf16.f32 	%rs148, %r159;
+	// begin inline asm
+	@%p1 st.global.b16 [ %rd436 + 0 ], { %rs141 };
+	// end inline asm
+	// begin inline asm
+	@%p2 st.global.b16 [ %rd437 + 0 ], { %rs142 };
+	// end inline asm
+	// begin inline asm
+	@%p3 st.global.b16 [ %rd438 + 0 ], { %rs143 };
+	// end inline asm
+	// begin inline asm
+	@%p4 st.global.b16 [ %rd439 + 0 ], { %rs144 };
+	// end inline asm
+	// begin inline asm
+	@%p5 st.global.b16 [ %rd440 + 0 ], { %rs145 };
+	// end inline asm
+	// begin inline asm
+	@%p6 st.global.b16 [ %rd441 + 0 ], { %rs146 };
+	// end inline asm
+	// begin inline asm
+	@%p7 st.global.b16 [ %rd442 + 0 ], { %rs147 };
+	// end inline asm
+	// begin inline asm
+	@%p8 st.global.b16 [ %rd443 + 0 ], { %rs148 };
+	// end inline asm
+	.loc	1 56 4                          // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:56:4
+	ret;
+$L__BB0_49:
+	.loc	1 32 62                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:32:62
+	{ // callseq 0, 0
+	.param .b64 	param0;
+	.param .b64 	param1;
+	.param .b32 	param2;
+	.param .b64 	param3;
+	.param .b64 	param4;
+	mov.b64 	%rd228, assertFunc_0;
+	cvta.global.u64 	%rd229, %rd228;
+	st.param.b64 	[param3], %rd229;
+	mov.b64 	%rd230, assertFile_0;
+	cvta.global.u64 	%rd231, %rd230;
+	st.param.b64 	[param1], %rd231;
+	mov.b64 	%rd232, assertMessage_0;
+	cvta.global.u64 	%rd233, %rd232;
+	st.param.b64 	[param0], %rd233;
+	st.param.b64 	[param4], 1;
+	st.param.b32 	[param2], 32;
+	call.uni __assertfail, (param0, param1, param2, param3, param4);
+	} // callseq 0
+	trap;
+$L__BB0_51:
+	.loc	1 52 64                         // cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py:52:64
+	{ // callseq 1, 0
+	.param .b64 	param0;
+	.param .b64 	param1;
+	.param .b32 	param2;
+	.param .b64 	param3;
+	.param .b64 	param4;
+	mov.b64 	%rd406, assertFunc_1;
+	cvta.global.u64 	%rd407, %rd406;
+	st.param.b64 	[param3], %rd407;
+	mov.b64 	%rd408, assertFile_1;
+	cvta.global.u64 	%rd409, %rd408;
+	st.param.b64 	[param1], %rd409;
+	mov.b64 	%rd410, assertMessage_1;
+	cvta.global.u64 	%rd411, %rd410;
+	st.param.b64 	[param0], %rd411;
+	st.param.b64 	[param4], 1;
+	st.param.b32 	[param2], 52;
+	call.uni __assertfail, (param0, param1, param2, param3, param4);
+	} // callseq 1
+	trap;
+$L__tmp1:
+$L__func_end0:
+                                        // -- End function
+}
+	.file	1 "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py"
+	.section	.debug_abbrev
+	{
+.b8 1                                   // Abbreviation Code
+.b8 17                                  // DW_TAG_compile_unit
+.b8 0                                   // DW_CHILDREN_no
+.b8 37                                  // DW_AT_producer
+.b8 8                                   // DW_FORM_string
+.b8 19                                  // DW_AT_language
+.b8 5                                   // DW_FORM_data2
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 16                                  // DW_AT_stmt_list
+.b8 6                                   // DW_FORM_data4
+.b8 27                                  // DW_AT_comp_dir
+.b8 8                                   // DW_FORM_string
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 0                                   // EOM(3)
+	}
+	.section	.debug_info
+	{
+.b32 135                                // Length of Unit
+.b8 2                                   // DWARF version number
+.b8 0
+.b32 .debug_abbrev                      // Offset Into Abbrev. Section
+.b8 8                                   // Address Size (in bytes)
+.b8 1                                   // Abbrev [1] 0xb:0x80 DW_TAG_compile_unit
+.b8 116                                 // DW_AT_producer
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 0
+.b8 2                                   // DW_AT_language
+.b8 0
+.b8 99                                  // DW_AT_name
+.b8 97
+.b8 108
+.b8 50
+.b8 114
+.b8 52
+.b8 116
+.b8 102
+.b8 121
+.b8 119
+.b8 54
+.b8 103
+.b8 105
+.b8 99
+.b8 51
+.b8 103
+.b8 103
+.b8 113
+.b8 121
+.b8 117
+.b8 100
+.b8 51
+.b8 110
+.b8 117
+.b8 102
+.b8 110
+.b8 97
+.b8 106
+.b8 120
+.b8 54
+.b8 120
+.b8 97
+.b8 117
+.b8 50
+.b8 107
+.b8 111
+.b8 105
+.b8 101
+.b8 111
+.b8 105
+.b8 116
+.b8 120
+.b8 54
+.b8 122
+.b8 103
+.b8 52
+.b8 119
+.b8 115
+.b8 105
+.b8 111
+.b8 122
+.b8 109
+.b8 46
+.b8 112
+.b8 121
+.b8 0
+.b32 .debug_line                        // DW_AT_stmt_list
+.b8 47                                  // DW_AT_comp_dir
+.b8 119
+.b8 111
+.b8 114
+.b8 107
+.b8 115
+.b8 112
+.b8 97
+.b8 99
+.b8 101
+.b8 47
+.b8 104
+.b8 97
+.b8 110
+.b8 114
+.b8 117
+.b8 105
+.b8 47
+.b8 83
+.b8 112
+.b8 101
+.b8 99
+.b8 70
+.b8 111
+.b8 114
+.b8 103
+.b8 101
+.b8 45
+.b8 101
+.b8 120
+.b8 116
+.b8 47
+.b8 99
+.b8 97
+.b8 99
+.b8 104
+.b8 101
+.b8 47
+.b8 99
+.b8 111
+.b8 109
+.b8 112
+.b8 105
+.b8 108
+.b8 101
+.b8 100
+.b8 95
+.b8 107
+.b8 101
+.b8 114
+.b8 110
+.b8 101
+.b8 108
+.b8 115
+.b8 47
+.b8 97
+.b8 108
+.b8 0
+	}
+	.section	.debug_macinfo	{	}

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.source

@@ -0,0 +1,299 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":18:0)
+#loc78 = loc("in_ptr0"(#loc))
+#loc79 = loc("in_ptr1"(#loc))
+#loc80 = loc("in_ptr2"(#loc))
+#loc81 = loc("in_ptr3"(#loc))
+#loc82 = loc("out_ptr0"(#loc))
+#loc83 = loc("ks0"(#loc))
+#loc84 = loc("ks1"(#loc))
+#loc85 = loc("ks2"(#loc))
+#loc86 = loc("ks3"(#loc))
+#loc87 = loc("ks4"(#loc))
+#loc88 = loc("xnumel"(#loc))
+module {
+  tt.func public @triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0(%in_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %in_ptr1: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("in_ptr1"(#loc)), %in_ptr2: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr2"(#loc)), %in_ptr3: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr3"(#loc)), %out_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %ks2: i64 loc("ks2"(#loc)), %ks3: i64 loc("ks3"(#loc)), %ks4: i64 loc("ks4"(#loc)), %xnumel: i32 loc("xnumel"(#loc))) attributes {noinline = false} {
+    %xoffset = tt.get_program_id x : i32 loc(#loc89)
+    %xoffset_0 = arith.constant 1024 : i32 loc(#loc90)
+    %xoffset_1 = arith.constant 1024 : i32 loc(#loc90)
+    %xoffset_2 = arith.muli %xoffset, %xoffset_1 : i32 loc(#loc90)
+    %xindex = tt.make_range {end = 1024 : i32, start = 0 : i32} : tensor<1024xi32> loc(#loc91)
+    %xindex_3 = tt.splat %xoffset_2 : i32 -> tensor<1024xi32> loc(#loc92)
+    %xindex_4 = arith.addi %xindex_3, %xindex : tensor<1024xi32> loc(#loc92)
+    %xmask = tt.splat %xnumel : i32 -> tensor<1024xi32> loc(#loc93)
+    %xmask_5 = arith.cmpi slt, %xindex_4, %xmask : tensor<1024xi32> loc(#loc93)
+    %x2 = arith.extsi %xindex_4 : tensor<1024xi32> to tensor<1024xi64> loc(#loc94)
+    %x2_6 = tt.splat %ks0 : i64 -> tensor<1024xi64> loc(#loc94)
+    %x2_7 = arith.divsi %x2, %x2_6 : tensor<1024xi64> loc(#loc94)
+    %x2_8 = tt.splat %ks1 : i64 -> tensor<1024xi64> loc(#loc95)
+    %x2_9 = arith.remsi %x2_7, %x2_8 : tensor<1024xi64> loc(#loc95)
+    %x0 = arith.extsi %xindex_4 : tensor<1024xi32> to tensor<1024xi64> loc(#loc96)
+    %x0_10 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc96)
+    %x0_11 = arith.remsi %x0, %x0_10 : tensor<1024xi64> loc(#loc96)
+    %x5 = arith.extsi %xindex_4 : tensor<1024xi32> to tensor<1024xi64> loc(#loc97)
+    %x5_12 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc97)
+    %x5_13 = arith.divsi %x5, %x5_12 : tensor<1024xi64> loc(#loc97)
+    %tmp0 = tt.splat %in_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc98)
+    %tmp0_14 = tt.addptr %tmp0, %xindex_4 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi32> loc(#loc98)
+    %tmp0_15 = tt.load %tmp0_14, %xmask_5 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc99)
+    %tmp0_16 = arith.extf %tmp0_15 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc100)
+    %tmp1 = tt.splat %in_ptr1 : !tt.ptr<i64> -> tensor<1024x!tt.ptr<i64>> loc(#loc101)
+    %tmp1_17 = tt.addptr %tmp1, %x2_9 : tensor<1024x!tt.ptr<i64>>, tensor<1024xi64> loc(#loc101)
+    %tmp1_18 = tt.load %tmp1_17, %xmask_5 evictionPolicy = evict_last : tensor<1024x!tt.ptr<i64>> loc(#loc102)
+    %tmp3 = tt.splat %ks2 : i64 -> tensor<1024xi64> loc(#loc103)
+    %tmp3_19 = arith.addi %tmp1_18, %tmp3 : tensor<1024xi64> loc(#loc103)
+    %tmp4 = arith.constant 0 : i32 loc(#loc104)
+    %tmp4_20 = arith.extsi %tmp4 : i32 to i64 loc(#loc104)
+    %tmp4_21 = tt.splat %tmp4_20 : i64 -> tensor<1024xi64> loc(#loc104)
+    %tmp4_22 = arith.cmpi slt, %tmp1_18, %tmp4_21 : tensor<1024xi64> loc(#loc104)
+    %tmp5 = arith.select %tmp4_22, %tmp3_19, %tmp1_18 : tensor<1024xi1>, tensor<1024xi64> loc(#loc105)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc18)
+    %0 = arith.extsi %c0_i32 : i32 to i64 loc(#loc18)
+    %1 = tt.splat %0 : i64 -> tensor<1024xi64> loc(#loc18)
+    %2 = arith.cmpi sle, %1, %tmp5 : tensor<1024xi64> loc(#loc18)
+    %3 = tt.splat %ks2 : i64 -> tensor<1024xi64> loc(#loc19)
+    %4 = arith.cmpi slt, %tmp5, %3 : tensor<1024xi64> loc(#loc19)
+    %5 = arith.andi %2, %4 : tensor<1024xi1> loc(#loc20)
+    %true = arith.constant true loc(#loc21)
+    %cst = arith.constant dense<true> : tensor<1024xi1> loc(#loc21)
+    %6 = arith.xori %xmask_5, %cst : tensor<1024xi1> loc(#loc21)
+    %7 = arith.ori %5, %6 : tensor<1024xi1> loc(#loc22)
+    tt.assert %7, "index out of bounds: 0 <= tmp5 < ks2" : tensor<1024xi1> loc(#loc23)
+    %tmp7 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc106)
+    %tmp7_23 = arith.muli %tmp7, %tmp5 : tensor<1024xi64> loc(#loc106)
+    %tmp7_24 = arith.addi %x0_11, %tmp7_23 : tensor<1024xi64> loc(#loc107)
+    %tmp7_25 = tt.splat %in_ptr2 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc108)
+    %tmp7_26 = tt.addptr %tmp7_25, %tmp7_24 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc108)
+    %tmp7_27 = tt.load %tmp7_26, %xmask_5 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc109)
+    %tmp7_28 = arith.extf %tmp7_27 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc110)
+    %tmp8 = arith.mulf %tmp0_16, %tmp7_28 : tensor<1024xf32> loc(#loc111)
+    %tmp10 = arith.constant 0 : i64 loc(#loc112)
+    %tmp10_29 = arith.constant dense<0> : tensor<1xi64> loc(#loc112)
+    %tmp11 = arith.constant dense<0> : tensor<1024xi64> loc(#loc113)
+    %tmp11_30 = arith.cmpi sge, %x0_11, %tmp11 : tensor<1024xi64> loc(#loc113)
+    %tmp12 = arith.constant 2 : i32 loc(#loc114)
+    %tmp12_31 = arith.constant 2 : i64 loc(#loc114)
+    %tmp12_32 = arith.divsi %ks3, %tmp12_31 : i64 loc(#loc114)
+    %tmp12_33 = arith.constant -1 : i32 loc(#loc115)
+    %tmp12_34 = arith.constant -1 : i64 loc(#loc115)
+    %tmp12_35 = arith.muli %tmp12_34, %tmp12_32 : i64 loc(#loc115)
+    %tmp12_36 = arith.addi %ks3, %tmp12_35 : i64 loc(#loc116)
+    %tmp13 = tt.splat %tmp12_36 : i64 -> tensor<1024xi64> loc(#loc117)
+    %tmp13_37 = arith.cmpi slt, %x0_11, %tmp13 : tensor<1024xi64> loc(#loc117)
+    %tmp14 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc118)
+    %tmp14_38 = arith.muli %tmp14, %x5_13 : tensor<1024xi64> loc(#loc118)
+    %tmp14_39 = arith.constant 2 : i32 loc(#loc119)
+    %tmp14_40 = arith.constant 2 : i64 loc(#loc119)
+    %tmp14_41 = arith.divsi %ks3, %tmp14_40 : i64 loc(#loc119)
+    %tmp14_42 = tt.splat %tmp14_41 : i64 -> tensor<1024xi64> loc(#loc120)
+    %tmp14_43 = arith.addi %tmp14_38, %tmp14_42 : tensor<1024xi64> loc(#loc120)
+    %tmp14_44 = arith.addi %tmp14_43, %x0_11 : tensor<1024xi64> loc(#loc121)
+    %tmp14_45 = tt.splat %in_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc122)
+    %tmp14_46 = tt.addptr %tmp14_45, %tmp14_44 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc122)
+    %tmp14_47 = arith.andi %tmp13_37, %xmask_5 : tensor<1024xi1> loc(#loc123)
+    %tmp14_48 = arith.constant 0.000000e+00 : f32 loc(#loc124)
+    %tmp14_49 = arith.constant dense<0.000000e+00> : tensor<1024xf32> loc(#loc124)
+    %tmp14_50 = arith.truncf %tmp14_49 : tensor<1024xf32> to tensor<1024xbf16> loc(#loc124)
+    %tmp14_51 = tt.load %tmp14_46, %tmp14_47, %tmp14_50 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc124)
+    %tmp14_52 = arith.extf %tmp14_51 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc125)
+    %tmp15 = arith.constant 0.000000e+00 : f32 loc(#loc126)
+    %tmp15_53 = arith.constant dense<0.000000e+00> : tensor<1024xf32> loc(#loc126)
+    %tmp15_54 = arith.subf %tmp15_53, %tmp14_52 : tensor<1024xf32> loc(#loc126)
+    %tmp16 = arith.constant 0.000000e+00 : f32 loc(#loc127)
+    %tmp16_55 = arith.constant dense<0.000000e+00> : tensor<1024xf32> loc(#loc127)
+    %tmp17 = arith.select %tmp13_37, %tmp15_54, %tmp16_55 : tensor<1024xi1>, tensor<1024xf32> loc(#loc128)
+    %tmp18 = tt.splat %tmp12_36 : i64 -> tensor<1024xi64> loc(#loc129)
+    %tmp18_56 = arith.cmpi sge, %x0_11, %tmp18 : tensor<1024xi64> loc(#loc129)
+    %tmp20 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc130)
+    %tmp20_57 = arith.cmpi slt, %x0_11, %tmp20 : tensor<1024xi64> loc(#loc130)
+    %tmp21 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc131)
+    %tmp21_58 = arith.muli %tmp21, %x5_13 : tensor<1024xi64> loc(#loc131)
+    %tmp21_59 = arith.constant -1 : i32 loc(#loc132)
+    %tmp21_60 = arith.constant -1 : i64 loc(#loc132)
+    %tmp21_61 = arith.muli %tmp21_60, %ks3 : i64 loc(#loc132)
+    %tmp21_62 = tt.splat %tmp21_61 : i64 -> tensor<1024xi64> loc(#loc133)
+    %tmp21_63 = arith.addi %x0_11, %tmp21_62 : tensor<1024xi64> loc(#loc133)
+    %tmp21_64 = arith.constant 2 : i32 loc(#loc134)
+    %tmp21_65 = arith.constant 2 : i64 loc(#loc134)
+    %tmp21_66 = arith.divsi %ks3, %tmp21_65 : i64 loc(#loc134)
+    %tmp21_67 = tt.splat %tmp21_66 : i64 -> tensor<1024xi64> loc(#loc135)
+    %tmp21_68 = arith.addi %tmp21_63, %tmp21_67 : tensor<1024xi64> loc(#loc135)
+    %tmp21_69 = arith.addi %tmp21_58, %tmp21_68 : tensor<1024xi64> loc(#loc136)
+    %tmp21_70 = tt.splat %in_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc137)
+    %tmp21_71 = tt.addptr %tmp21_70, %tmp21_69 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc137)
+    %tmp21_72 = arith.andi %tmp18_56, %xmask_5 : tensor<1024xi1> loc(#loc138)
+    %tmp21_73 = arith.constant 0.000000e+00 : f32 loc(#loc139)
+    %tmp21_74 = arith.constant dense<0.000000e+00> : tensor<1024xf32> loc(#loc139)
+    %tmp21_75 = arith.truncf %tmp21_74 : tensor<1024xf32> to tensor<1024xbf16> loc(#loc139)
+    %tmp21_76 = tt.load %tmp21_71, %tmp21_72, %tmp21_75 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc139)
+    %tmp21_77 = arith.extf %tmp21_76 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc140)
+    %tmp22 = arith.select %tmp13_37, %tmp17, %tmp21_77 : tensor<1024xi1>, tensor<1024xf32> loc(#loc141)
+    %tmp24 = tt.splat %ks4 : i64 -> tensor<1024xi64> loc(#loc142)
+    %tmp24_78 = arith.addi %tmp1_18, %tmp24 : tensor<1024xi64> loc(#loc142)
+    %tmp25 = arith.select %tmp4_22, %tmp24_78, %tmp1_18 : tensor<1024xi1>, tensor<1024xi64> loc(#loc143)
+    %c0_i32_79 = arith.constant 0 : i32 loc(#loc62)
+    %8 = arith.extsi %c0_i32_79 : i32 to i64 loc(#loc62)
+    %9 = tt.splat %8 : i64 -> tensor<1024xi64> loc(#loc62)
+    %10 = arith.cmpi sle, %9, %tmp25 : tensor<1024xi64> loc(#loc62)
+    %11 = tt.splat %ks4 : i64 -> tensor<1024xi64> loc(#loc63)
+    %12 = arith.cmpi slt, %tmp25, %11 : tensor<1024xi64> loc(#loc63)
+    %13 = arith.andi %10, %12 : tensor<1024xi1> loc(#loc64)
+    %true_80 = arith.constant true loc(#loc65)
+    %cst_81 = arith.constant dense<true> : tensor<1024xi1> loc(#loc65)
+    %14 = arith.xori %xmask_5, %cst_81 : tensor<1024xi1> loc(#loc65)
+    %15 = arith.ori %13, %14 : tensor<1024xi1> loc(#loc66)
+    tt.assert %15, "index out of bounds: 0 <= tmp25 < ks4" : tensor<1024xi1> loc(#loc67)
+    %tmp27 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc144)
+    %tmp27_82 = arith.muli %tmp27, %tmp25 : tensor<1024xi64> loc(#loc144)
+    %tmp27_83 = arith.addi %x0_11, %tmp27_82 : tensor<1024xi64> loc(#loc145)
+    %tmp27_84 = tt.splat %in_ptr3 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc146)
+    %tmp27_85 = tt.addptr %tmp27_84, %tmp27_83 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc146)
+    %tmp27_86 = tt.load %tmp27_85, %xmask_5 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc147)
+    %tmp27_87 = arith.extf %tmp27_86 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc148)
+    %tmp28 = arith.mulf %tmp22, %tmp27_87 : tensor<1024xf32> loc(#loc149)
+    %tmp29 = arith.addf %tmp8, %tmp28 : tensor<1024xf32> loc(#loc150)
+    %16 = tt.splat %out_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc75)
+    %17 = tt.addptr %16, %xindex_4 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi32> loc(#loc75)
+    %18 = arith.truncf %tmp29 : tensor<1024xf32> to tensor<1024xbf16> loc(#loc76)
+    tt.store %17, %18, %xmask_5 : tensor<1024x!tt.ptr<bf16>> loc(#loc76)
+    tt.return loc(#loc77)
+  } loc(#loc)
+} loc(#loc)
+#loc1 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:28)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:33)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:36)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:23)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":21:21)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:28)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":24:19)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":25:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:30)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:35)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:75)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:30)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:35)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":29:18)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":30:18)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":31:32)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:28)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:44)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:37)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:54)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:52)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:62)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:39)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:35)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:30)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:46)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:86)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":34:18)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":36:28)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":37:20)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:31)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:24)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:18)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":39:19)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:35)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:48)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:41)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:54)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:31)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:68)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:60)
+#loc43 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:119)
+#loc44 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":41:13)
+#loc45 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":42:38)
+#loc46 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":43:35)
+#loc47 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":44:20)
+#loc48 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":46:19)
+#loc49 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:35)
+#loc50 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:52)
+#loc51 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:47)
+#loc52 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:67)
+#loc53 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:60)
+#loc54 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:41)
+#loc55 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:31)
+#loc56 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:81)
+#loc57 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:73)
+#loc58 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:132)
+#loc59 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":48:35)
+#loc60 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":50:19)
+#loc61 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":51:34)
+#loc62 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:28)
+#loc63 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:46)
+#loc64 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:38)
+#loc65 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:56)
+#loc66 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:54)
+#loc67 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:64)
+#loc68 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:40)
+#loc69 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:36)
+#loc70 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:31)
+#loc71 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:48)
+#loc72 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:88)
+#loc73 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":54:20)
+#loc74 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":55:19)
+#loc75 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:25)
+#loc76 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:37)
+#loc77 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:4)
+#loc89 = loc("xoffset"(#loc1))
+#loc90 = loc("xoffset"(#loc2))
+#loc91 = loc("xindex"(#loc3))
+#loc92 = loc("xindex"(#loc4))
+#loc93 = loc("xmask"(#loc5))
+#loc94 = loc("x2"(#loc6))
+#loc95 = loc("x2"(#loc7))
+#loc96 = loc("x0"(#loc8))
+#loc97 = loc("x5"(#loc9))
+#loc98 = loc("tmp0"(#loc10))
+#loc99 = loc("tmp0"(#loc11))
+#loc100 = loc("tmp0"(#loc12))
+#loc101 = loc("tmp1"(#loc13))
+#loc102 = loc("tmp1"(#loc14))
+#loc103 = loc("tmp3"(#loc15))
+#loc104 = loc("tmp4"(#loc16))
+#loc105 = loc("tmp5"(#loc17))
+#loc106 = loc("tmp7"(#loc24))
+#loc107 = loc("tmp7"(#loc25))
+#loc108 = loc("tmp7"(#loc26))
+#loc109 = loc("tmp7"(#loc27))
+#loc110 = loc("tmp7"(#loc28))
+#loc111 = loc("tmp8"(#loc29))
+#loc112 = loc("tmp10"(#loc30))
+#loc113 = loc("tmp11"(#loc31))
+#loc114 = loc("tmp12"(#loc32))
+#loc115 = loc("tmp12"(#loc33))
+#loc116 = loc("tmp12"(#loc34))
+#loc117 = loc("tmp13"(#loc35))
+#loc118 = loc("tmp14"(#loc36))
+#loc119 = loc("tmp14"(#loc37))
+#loc120 = loc("tmp14"(#loc38))
+#loc121 = loc("tmp14"(#loc39))
+#loc122 = loc("tmp14"(#loc40))
+#loc123 = loc("tmp14"(#loc41))
+#loc124 = loc("tmp14"(#loc42))
+#loc125 = loc("tmp14"(#loc43))
+#loc126 = loc("tmp15"(#loc44))
+#loc127 = loc("tmp16"(#loc45))
+#loc128 = loc("tmp17"(#loc46))
+#loc129 = loc("tmp18"(#loc47))
+#loc130 = loc("tmp20"(#loc48))
+#loc131 = loc("tmp21"(#loc49))
+#loc132 = loc("tmp21"(#loc50))
+#loc133 = loc("tmp21"(#loc51))
+#loc134 = loc("tmp21"(#loc52))
+#loc135 = loc("tmp21"(#loc53))
+#loc136 = loc("tmp21"(#loc54))
+#loc137 = loc("tmp21"(#loc55))
+#loc138 = loc("tmp21"(#loc56))
+#loc139 = loc("tmp21"(#loc57))
+#loc140 = loc("tmp21"(#loc58))
+#loc141 = loc("tmp22"(#loc59))
+#loc142 = loc("tmp24"(#loc60))
+#loc143 = loc("tmp25"(#loc61))
+#loc144 = loc("tmp27"(#loc68))
+#loc145 = loc("tmp27"(#loc69))
+#loc146 = loc("tmp27"(#loc70))
+#loc147 = loc("tmp27"(#loc71))
+#loc148 = loc("tmp27"(#loc72))
+#loc149 = loc("tmp28"(#loc73))
+#loc150 = loc("tmp29"(#loc74))

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttgir

@@ -0,0 +1,232 @@
+#blocked = #ttg.blocked<{sizePerThread = [8], threadsPerWarp = [32], warpsPerCTA = [4], order = [0]}>
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":18:0)
+#loc70 = loc("in_ptr0"(#loc))
+#loc71 = loc("in_ptr1"(#loc))
+#loc72 = loc("in_ptr2"(#loc))
+#loc73 = loc("in_ptr3"(#loc))
+#loc74 = loc("out_ptr0"(#loc))
+#loc75 = loc("ks0"(#loc))
+#loc76 = loc("ks1"(#loc))
+#loc77 = loc("ks2"(#loc))
+#loc78 = loc("ks3"(#loc))
+#loc79 = loc("ks4"(#loc))
+#loc80 = loc("xnumel"(#loc))
+module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 4 : i32, ttg.target = "cuda:90", "ttg.threads-per-warp" = 32 : i32} {
+  tt.func public @triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0(%in_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %in_ptr1: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("in_ptr1"(#loc)), %in_ptr2: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr2"(#loc)), %in_ptr3: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr3"(#loc)), %out_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %ks2: i64 loc("ks2"(#loc)), %ks3: i64 loc("ks3"(#loc)), %ks4: i64 loc("ks4"(#loc)), %xnumel: i32 loc("xnumel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<true> : tensor<1024xi1, #blocked> loc(#loc1)
+    %c1024_i32 = arith.constant 1024 : i32 loc(#loc1)
+    %cst_0 = arith.constant dense<0.000000e+00> : tensor<1024xbf16, #blocked> loc(#loc1)
+    %c2_i64 = arith.constant 2 : i64 loc(#loc1)
+    %c-1_i64 = arith.constant -1 : i64 loc(#loc1)
+    %cst_1 = arith.constant dense<0> : tensor<1024xi64, #blocked> loc(#loc1)
+    %cst_2 = arith.constant dense<0.000000e+00> : tensor<1024xf32, #blocked> loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc81)
+    %xoffset_3 = arith.muli %xoffset, %c1024_i32 : i32 loc(#loc82)
+    %xindex = tt.make_range {end = 1024 : i32, start = 0 : i32} : tensor<1024xi32, #blocked> loc(#loc83)
+    %xindex_4 = tt.splat %xoffset_3 : i32 -> tensor<1024xi32, #blocked> loc(#loc84)
+    %xindex_5 = arith.addi %xindex_4, %xindex : tensor<1024xi32, #blocked> loc(#loc84)
+    %xmask = tt.splat %xnumel : i32 -> tensor<1024xi32, #blocked> loc(#loc85)
+    %xmask_6 = arith.cmpi slt, %xindex_5, %xmask : tensor<1024xi32, #blocked> loc(#loc85)
+    %x2 = arith.extsi %xindex_5 : tensor<1024xi32, #blocked> to tensor<1024xi64, #blocked> loc(#loc86)
+    %x2_7 = tt.splat %ks0 : i64 -> tensor<1024xi64, #blocked> loc(#loc86)
+    %x2_8 = arith.divsi %x2, %x2_7 : tensor<1024xi64, #blocked> loc(#loc86)
+    %x2_9 = tt.splat %ks1 : i64 -> tensor<1024xi64, #blocked> loc(#loc87)
+    %x2_10 = arith.remsi %x2_8, %x2_9 : tensor<1024xi64, #blocked> loc(#loc87)
+    %x0 = tt.splat %ks3 : i64 -> tensor<1024xi64, #blocked> loc(#loc88)
+    %x0_11 = arith.remsi %x2, %x0 : tensor<1024xi64, #blocked> loc(#loc88)
+    %x5 = arith.divsi %x2, %x0 : tensor<1024xi64, #blocked> loc(#loc89)
+    %tmp0 = tt.splat %in_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc90)
+    %tmp0_12 = tt.addptr %tmp0, %xindex_5 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi32, #blocked> loc(#loc90)
+    %tmp0_13 = tt.load %tmp0_12, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc91)
+    %tmp0_14 = arith.extf %tmp0_13 : tensor<1024xbf16, #blocked> to tensor<1024xf32, #blocked> loc(#loc92)
+    %tmp1 = tt.splat %in_ptr1 : !tt.ptr<i64> -> tensor<1024x!tt.ptr<i64>, #blocked> loc(#loc93)
+    %tmp1_15 = tt.addptr %tmp1, %x2_10 : tensor<1024x!tt.ptr<i64>, #blocked>, tensor<1024xi64, #blocked> loc(#loc93)
+    %tmp1_16 = tt.load %tmp1_15, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<i64>, #blocked> loc(#loc94)
+    %tmp3 = tt.splat %ks2 : i64 -> tensor<1024xi64, #blocked> loc(#loc95)
+    %tmp3_17 = arith.addi %tmp1_16, %tmp3 : tensor<1024xi64, #blocked> loc(#loc95)
+    %tmp4 = arith.cmpi slt, %tmp1_16, %cst_1 : tensor<1024xi64, #blocked> loc(#loc96)
+    %tmp5 = arith.select %tmp4, %tmp3_17, %tmp1_16 : tensor<1024xi1, #blocked>, tensor<1024xi64, #blocked> loc(#loc97)
+    %0 = arith.cmpi sge, %tmp5, %cst_1 : tensor<1024xi64, #blocked> loc(#loc19)
+    %1 = arith.cmpi slt, %tmp5, %tmp3 : tensor<1024xi64, #blocked> loc(#loc20)
+    %2 = arith.andi %0, %1 : tensor<1024xi1, #blocked> loc(#loc21)
+    %3 = arith.xori %xmask_6, %cst : tensor<1024xi1, #blocked> loc(#loc22)
+    %4 = arith.ori %2, %3 : tensor<1024xi1, #blocked> loc(#loc23)
+    tt.assert %4, "index out of bounds: 0 <= tmp5 < ks2" : tensor<1024xi1, #blocked> loc(#loc24)
+    %tmp7 = arith.muli %x0, %tmp5 : tensor<1024xi64, #blocked> loc(#loc98)
+    %tmp7_18 = arith.addi %x0_11, %tmp7 : tensor<1024xi64, #blocked> loc(#loc99)
+    %tmp7_19 = tt.splat %in_ptr2 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc100)
+    %tmp7_20 = tt.addptr %tmp7_19, %tmp7_18 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi64, #blocked> loc(#loc100)
+    %tmp7_21 = tt.load %tmp7_20, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc101)
+    %tmp7_22 = arith.extf %tmp7_21 : tensor<1024xbf16, #blocked> to tensor<1024xf32, #blocked> loc(#loc102)
+    %tmp8 = arith.mulf %tmp0_14, %tmp7_22 : tensor<1024xf32, #blocked> loc(#loc103)
+    %tmp12 = arith.divsi %ks3, %c2_i64 : i64 loc(#loc104)
+    %tmp12_23 = arith.subi %ks3, %tmp12 : i64 loc(#loc105)
+    %tmp13 = tt.splat %tmp12_23 : i64 -> tensor<1024xi64, #blocked> loc(#loc106)
+    %tmp13_24 = arith.cmpi slt, %x0_11, %tmp13 : tensor<1024xi64, #blocked> loc(#loc106)
+    %tmp14 = arith.muli %x0, %x5 : tensor<1024xi64, #blocked> loc(#loc107)
+    %tmp14_25 = tt.splat %tmp12 : i64 -> tensor<1024xi64, #blocked> loc(#loc108)
+    %tmp14_26 = arith.addi %tmp14, %tmp14_25 : tensor<1024xi64, #blocked> loc(#loc108)
+    %tmp14_27 = arith.addi %tmp14_26, %x0_11 : tensor<1024xi64, #blocked> loc(#loc109)
+    %tmp14_28 = tt.addptr %tmp0, %tmp14_27 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi64, #blocked> loc(#loc110)
+    %tmp14_29 = arith.andi %tmp13_24, %xmask_6 : tensor<1024xi1, #blocked> loc(#loc111)
+    %tmp14_30 = tt.load %tmp14_28, %tmp14_29, %cst_0 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc112)
+    %tmp14_31 = arith.extf %tmp14_30 : tensor<1024xbf16, #blocked> to tensor<1024xf32, #blocked> loc(#loc113)
+    %tmp15 = arith.subf %cst_2, %tmp14_31 : tensor<1024xf32, #blocked> loc(#loc114)
+    %tmp18 = arith.cmpi sge, %x0_11, %tmp13 : tensor<1024xi64, #blocked> loc(#loc115)
+    %tmp21 = arith.muli %ks3, %c-1_i64 : i64 loc(#loc116)
+    %tmp21_32 = tt.splat %tmp21 : i64 -> tensor<1024xi64, #blocked> loc(#loc117)
+    %tmp21_33 = arith.addi %x0_11, %tmp21_32 : tensor<1024xi64, #blocked> loc(#loc117)
+    %tmp21_34 = arith.addi %tmp21_33, %tmp14_25 : tensor<1024xi64, #blocked> loc(#loc118)
+    %tmp21_35 = arith.addi %tmp14, %tmp21_34 : tensor<1024xi64, #blocked> loc(#loc119)
+    %tmp21_36 = tt.addptr %tmp0, %tmp21_35 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi64, #blocked> loc(#loc120)
+    %tmp21_37 = arith.andi %tmp18, %xmask_6 : tensor<1024xi1, #blocked> loc(#loc121)
+    %tmp21_38 = tt.load %tmp21_36, %tmp21_37, %cst_0 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc122)
+    %tmp21_39 = arith.extf %tmp21_38 : tensor<1024xbf16, #blocked> to tensor<1024xf32, #blocked> loc(#loc123)
+    %tmp22 = arith.select %tmp13_24, %tmp15, %tmp21_39 : tensor<1024xi1, #blocked>, tensor<1024xf32, #blocked> loc(#loc135)
+    %tmp24 = tt.splat %ks4 : i64 -> tensor<1024xi64, #blocked> loc(#loc126)
+    %tmp24_40 = arith.addi %tmp1_16, %tmp24 : tensor<1024xi64, #blocked> loc(#loc126)
+    %tmp25 = arith.select %tmp4, %tmp24_40, %tmp1_16 : tensor<1024xi1, #blocked>, tensor<1024xi64, #blocked> loc(#loc127)
+    %5 = arith.cmpi sge, %tmp25, %cst_1 : tensor<1024xi64, #blocked> loc(#loc55)
+    %6 = arith.cmpi slt, %tmp25, %tmp24 : tensor<1024xi64, #blocked> loc(#loc56)
+    %7 = arith.andi %5, %6 : tensor<1024xi1, #blocked> loc(#loc57)
+    %8 = arith.ori %7, %3 : tensor<1024xi1, #blocked> loc(#loc58)
+    tt.assert %8, "index out of bounds: 0 <= tmp25 < ks4" : tensor<1024xi1, #blocked> loc(#loc59)
+    %tmp27 = arith.muli %x0, %tmp25 : tensor<1024xi64, #blocked> loc(#loc128)
+    %tmp27_41 = arith.addi %x0_11, %tmp27 : tensor<1024xi64, #blocked> loc(#loc129)
+    %tmp27_42 = tt.splat %in_ptr3 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc130)
+    %tmp27_43 = tt.addptr %tmp27_42, %tmp27_41 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi64, #blocked> loc(#loc130)
+    %tmp27_44 = tt.load %tmp27_43, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc131)
+    %tmp27_45 = arith.extf %tmp27_44 : tensor<1024xbf16, #blocked> to tensor<1024xf32, #blocked> loc(#loc132)
+    %tmp28 = arith.mulf %tmp22, %tmp27_45 : tensor<1024xf32, #blocked> loc(#loc133)
+    %tmp29 = arith.addf %tmp8, %tmp28 : tensor<1024xf32, #blocked> loc(#loc134)
+    %9 = tt.splat %out_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc67)
+    %10 = tt.addptr %9, %xindex_5 : tensor<1024x!tt.ptr<bf16>, #blocked>, tensor<1024xi32, #blocked> loc(#loc67)
+    %11 = arith.truncf %tmp29 : tensor<1024xf32, #blocked> to tensor<1024xbf16, #blocked> loc(#loc68)
+    tt.store %10, %11, %xmask_6 : tensor<1024x!tt.ptr<bf16>, #blocked> loc(#loc68)
+    tt.return loc(#loc69)
+  } loc(#loc)
+} loc(#loc)
+#loc1 = loc(unknown)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:36)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":21:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:21)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:28)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":24:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":25:19)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:30)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:35)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:75)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:30)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:35)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":29:18)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":30:18)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":31:32)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:28)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:44)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:37)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:54)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:52)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:62)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:39)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:35)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:30)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:46)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:86)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":34:18)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:31)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:18)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":39:19)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:35)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:41)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:54)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:31)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:68)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:60)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:119)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":41:13)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":44:20)
+#loc43 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:52)
+#loc44 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:47)
+#loc45 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:60)
+#loc46 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:41)
+#loc47 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:31)
+#loc48 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:81)
+#loc49 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:73)
+#loc50 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:132)
+#loc51 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":48:35)
+#loc52 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":43:35)
+#loc53 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":50:19)
+#loc54 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":51:34)
+#loc55 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:28)
+#loc56 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:46)
+#loc57 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:38)
+#loc58 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:54)
+#loc59 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:64)
+#loc60 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:40)
+#loc61 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:36)
+#loc62 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:31)
+#loc63 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:48)
+#loc64 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:88)
+#loc65 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":54:20)
+#loc66 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":55:19)
+#loc67 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:25)
+#loc68 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:37)
+#loc69 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:4)
+#loc81 = loc("xoffset"(#loc2))
+#loc82 = loc("xoffset"(#loc3))
+#loc83 = loc("xindex"(#loc4))
+#loc84 = loc("xindex"(#loc5))
+#loc85 = loc("xmask"(#loc6))
+#loc86 = loc("x2"(#loc7))
+#loc87 = loc("x2"(#loc8))
+#loc88 = loc("x0"(#loc9))
+#loc89 = loc("x5"(#loc10))
+#loc90 = loc("tmp0"(#loc11))
+#loc91 = loc("tmp0"(#loc12))
+#loc92 = loc("tmp0"(#loc13))
+#loc93 = loc("tmp1"(#loc14))
+#loc94 = loc("tmp1"(#loc15))
+#loc95 = loc("tmp3"(#loc16))
+#loc96 = loc("tmp4"(#loc17))
+#loc97 = loc("tmp5"(#loc18))
+#loc98 = loc("tmp7"(#loc25))
+#loc99 = loc("tmp7"(#loc26))
+#loc100 = loc("tmp7"(#loc27))
+#loc101 = loc("tmp7"(#loc28))
+#loc102 = loc("tmp7"(#loc29))
+#loc103 = loc("tmp8"(#loc30))
+#loc104 = loc("tmp12"(#loc31))
+#loc105 = loc("tmp12"(#loc32))
+#loc106 = loc("tmp13"(#loc33))
+#loc107 = loc("tmp14"(#loc34))
+#loc108 = loc("tmp14"(#loc35))
+#loc109 = loc("tmp14"(#loc36))
+#loc110 = loc("tmp14"(#loc37))
+#loc111 = loc("tmp14"(#loc38))
+#loc112 = loc("tmp14"(#loc39))
+#loc113 = loc("tmp14"(#loc40))
+#loc114 = loc("tmp15"(#loc41))
+#loc115 = loc("tmp18"(#loc42))
+#loc116 = loc("tmp21"(#loc43))
+#loc117 = loc("tmp21"(#loc44))
+#loc118 = loc("tmp21"(#loc45))
+#loc119 = loc("tmp21"(#loc46))
+#loc120 = loc("tmp21"(#loc47))
+#loc121 = loc("tmp21"(#loc48))
+#loc122 = loc("tmp21"(#loc49))
+#loc123 = loc("tmp21"(#loc50))
+#loc124 = loc("tmp22"(#loc51))
+#loc125 = loc("tmp17"(#loc52))
+#loc126 = loc("tmp24"(#loc53))
+#loc127 = loc("tmp25"(#loc54))
+#loc128 = loc("tmp27"(#loc60))
+#loc129 = loc("tmp27"(#loc61))
+#loc130 = loc("tmp27"(#loc62))
+#loc131 = loc("tmp27"(#loc63))
+#loc132 = loc("tmp27"(#loc64))
+#loc133 = loc("tmp28"(#loc65))
+#loc134 = loc("tmp29"(#loc66))
+#loc135 = loc(fused[#loc124, #loc125])

SpecForge-ext/cache/compiled_kernels/triton/0/2TU6ZCF6AOXLWQQED5J7FS5ZXMYK7TIOQ6T2MLB767275BROXJCA/triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0.ttir

@@ -0,0 +1,231 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":18:0)
+#loc70 = loc("in_ptr0"(#loc))
+#loc71 = loc("in_ptr1"(#loc))
+#loc72 = loc("in_ptr2"(#loc))
+#loc73 = loc("in_ptr3"(#loc))
+#loc74 = loc("out_ptr0"(#loc))
+#loc75 = loc("ks0"(#loc))
+#loc76 = loc("ks1"(#loc))
+#loc77 = loc("ks2"(#loc))
+#loc78 = loc("ks3"(#loc))
+#loc79 = loc("ks4"(#loc))
+#loc80 = loc("xnumel"(#loc))
+module {
+  tt.func public @triton_poi_fused_add_cat_index_mul_neg_slice_squeeze_unsqueeze_0(%in_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %in_ptr1: !tt.ptr<i64> {tt.divisibility = 16 : i32} loc("in_ptr1"(#loc)), %in_ptr2: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr2"(#loc)), %in_ptr3: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("in_ptr3"(#loc)), %out_ptr0: !tt.ptr<bf16> {tt.divisibility = 16 : i32} loc("out_ptr0"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %ks2: i64 loc("ks2"(#loc)), %ks3: i64 loc("ks3"(#loc)), %ks4: i64 loc("ks4"(#loc)), %xnumel: i32 loc("xnumel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<0.000000e+00> : tensor<1024xbf16> loc(#loc1)
+    %cst_0 = arith.constant dense<0> : tensor<1024xi64> loc(#loc1)
+    %cst_1 = arith.constant dense<0.000000e+00> : tensor<1024xf32> loc(#loc1)
+    %c-1_i64 = arith.constant -1 : i64 loc(#loc1)
+    %c2_i64 = arith.constant 2 : i64 loc(#loc1)
+    %cst_2 = arith.constant dense<true> : tensor<1024xi1> loc(#loc1)
+    %c1024_i32 = arith.constant 1024 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc81)
+    %xoffset_3 = arith.muli %xoffset, %c1024_i32 : i32 loc(#loc82)
+    %xindex = tt.make_range {end = 1024 : i32, start = 0 : i32} : tensor<1024xi32> loc(#loc83)
+    %xindex_4 = tt.splat %xoffset_3 : i32 -> tensor<1024xi32> loc(#loc84)
+    %xindex_5 = arith.addi %xindex_4, %xindex : tensor<1024xi32> loc(#loc84)
+    %xmask = tt.splat %xnumel : i32 -> tensor<1024xi32> loc(#loc85)
+    %xmask_6 = arith.cmpi slt, %xindex_5, %xmask : tensor<1024xi32> loc(#loc85)
+    %x2 = arith.extsi %xindex_5 : tensor<1024xi32> to tensor<1024xi64> loc(#loc86)
+    %x2_7 = tt.splat %ks0 : i64 -> tensor<1024xi64> loc(#loc86)
+    %x2_8 = arith.divsi %x2, %x2_7 : tensor<1024xi64> loc(#loc86)
+    %x2_9 = tt.splat %ks1 : i64 -> tensor<1024xi64> loc(#loc87)
+    %x2_10 = arith.remsi %x2_8, %x2_9 : tensor<1024xi64> loc(#loc87)
+    %x0 = tt.splat %ks3 : i64 -> tensor<1024xi64> loc(#loc88)
+    %x0_11 = arith.remsi %x2, %x0 : tensor<1024xi64> loc(#loc88)
+    %x5 = arith.divsi %x2, %x0 : tensor<1024xi64> loc(#loc89)
+    %tmp0 = tt.splat %in_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc90)
+    %tmp0_12 = tt.addptr %tmp0, %xindex_5 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi32> loc(#loc90)
+    %tmp0_13 = tt.load %tmp0_12, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc91)
+    %tmp0_14 = arith.extf %tmp0_13 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc92)
+    %tmp1 = tt.splat %in_ptr1 : !tt.ptr<i64> -> tensor<1024x!tt.ptr<i64>> loc(#loc93)
+    %tmp1_15 = tt.addptr %tmp1, %x2_10 : tensor<1024x!tt.ptr<i64>>, tensor<1024xi64> loc(#loc93)
+    %tmp1_16 = tt.load %tmp1_15, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<i64>> loc(#loc94)
+    %tmp3 = tt.splat %ks2 : i64 -> tensor<1024xi64> loc(#loc95)
+    %tmp3_17 = arith.addi %tmp1_16, %tmp3 : tensor<1024xi64> loc(#loc95)
+    %tmp4 = arith.cmpi slt, %tmp1_16, %cst_0 : tensor<1024xi64> loc(#loc96)
+    %tmp5 = arith.select %tmp4, %tmp3_17, %tmp1_16 : tensor<1024xi1>, tensor<1024xi64> loc(#loc97)
+    %0 = arith.cmpi sge, %tmp5, %cst_0 : tensor<1024xi64> loc(#loc19)
+    %1 = arith.cmpi slt, %tmp5, %tmp3 : tensor<1024xi64> loc(#loc20)
+    %2 = arith.andi %0, %1 : tensor<1024xi1> loc(#loc21)
+    %3 = arith.xori %xmask_6, %cst_2 : tensor<1024xi1> loc(#loc22)
+    %4 = arith.ori %2, %3 : tensor<1024xi1> loc(#loc23)
+    tt.assert %4, "index out of bounds: 0 <= tmp5 < ks2" : tensor<1024xi1> loc(#loc24)
+    %tmp7 = arith.muli %x0, %tmp5 : tensor<1024xi64> loc(#loc98)
+    %tmp7_18 = arith.addi %x0_11, %tmp7 : tensor<1024xi64> loc(#loc99)
+    %tmp7_19 = tt.splat %in_ptr2 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc100)
+    %tmp7_20 = tt.addptr %tmp7_19, %tmp7_18 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc100)
+    %tmp7_21 = tt.load %tmp7_20, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc101)
+    %tmp7_22 = arith.extf %tmp7_21 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc102)
+    %tmp8 = arith.mulf %tmp0_14, %tmp7_22 : tensor<1024xf32> loc(#loc103)
+    %tmp12 = arith.divsi %ks3, %c2_i64 : i64 loc(#loc104)
+    %tmp12_23 = arith.subi %ks3, %tmp12 : i64 loc(#loc105)
+    %tmp13 = tt.splat %tmp12_23 : i64 -> tensor<1024xi64> loc(#loc106)
+    %tmp13_24 = arith.cmpi slt, %x0_11, %tmp13 : tensor<1024xi64> loc(#loc106)
+    %tmp14 = arith.muli %x0, %x5 : tensor<1024xi64> loc(#loc107)
+    %tmp14_25 = tt.splat %tmp12 : i64 -> tensor<1024xi64> loc(#loc108)
+    %tmp14_26 = arith.addi %tmp14, %tmp14_25 : tensor<1024xi64> loc(#loc108)
+    %tmp14_27 = arith.addi %tmp14_26, %x0_11 : tensor<1024xi64> loc(#loc109)
+    %tmp14_28 = tt.addptr %tmp0, %tmp14_27 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc110)
+    %tmp14_29 = arith.andi %tmp13_24, %xmask_6 : tensor<1024xi1> loc(#loc111)
+    %tmp14_30 = tt.load %tmp14_28, %tmp14_29, %cst evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc112)
+    %tmp14_31 = arith.extf %tmp14_30 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc113)
+    %tmp15 = arith.subf %cst_1, %tmp14_31 : tensor<1024xf32> loc(#loc114)
+    %tmp18 = arith.cmpi sge, %x0_11, %tmp13 : tensor<1024xi64> loc(#loc115)
+    %tmp21 = arith.muli %ks3, %c-1_i64 : i64 loc(#loc116)
+    %tmp21_32 = tt.splat %tmp21 : i64 -> tensor<1024xi64> loc(#loc117)
+    %tmp21_33 = arith.addi %x0_11, %tmp21_32 : tensor<1024xi64> loc(#loc117)
+    %tmp21_34 = arith.addi %tmp21_33, %tmp14_25 : tensor<1024xi64> loc(#loc118)
+    %tmp21_35 = arith.addi %tmp14, %tmp21_34 : tensor<1024xi64> loc(#loc119)
+    %tmp21_36 = tt.addptr %tmp0, %tmp21_35 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc120)
+    %tmp21_37 = arith.andi %tmp18, %xmask_6 : tensor<1024xi1> loc(#loc121)
+    %tmp21_38 = tt.load %tmp21_36, %tmp21_37, %cst evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc122)
+    %tmp21_39 = arith.extf %tmp21_38 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc123)
+    %tmp22 = arith.select %tmp13_24, %tmp15, %tmp21_39 : tensor<1024xi1>, tensor<1024xf32> loc(#loc135)
+    %tmp24 = tt.splat %ks4 : i64 -> tensor<1024xi64> loc(#loc126)
+    %tmp24_40 = arith.addi %tmp1_16, %tmp24 : tensor<1024xi64> loc(#loc126)
+    %tmp25 = arith.select %tmp4, %tmp24_40, %tmp1_16 : tensor<1024xi1>, tensor<1024xi64> loc(#loc127)
+    %5 = arith.cmpi sge, %tmp25, %cst_0 : tensor<1024xi64> loc(#loc55)
+    %6 = arith.cmpi slt, %tmp25, %tmp24 : tensor<1024xi64> loc(#loc56)
+    %7 = arith.andi %5, %6 : tensor<1024xi1> loc(#loc57)
+    %8 = arith.ori %7, %3 : tensor<1024xi1> loc(#loc58)
+    tt.assert %8, "index out of bounds: 0 <= tmp25 < ks4" : tensor<1024xi1> loc(#loc59)
+    %tmp27 = arith.muli %x0, %tmp25 : tensor<1024xi64> loc(#loc128)
+    %tmp27_41 = arith.addi %x0_11, %tmp27 : tensor<1024xi64> loc(#loc129)
+    %tmp27_42 = tt.splat %in_ptr3 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc130)
+    %tmp27_43 = tt.addptr %tmp27_42, %tmp27_41 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi64> loc(#loc130)
+    %tmp27_44 = tt.load %tmp27_43, %xmask_6 evictionPolicy = evict_last : tensor<1024x!tt.ptr<bf16>> loc(#loc131)
+    %tmp27_45 = arith.extf %tmp27_44 : tensor<1024xbf16> to tensor<1024xf32> loc(#loc132)
+    %tmp28 = arith.mulf %tmp22, %tmp27_45 : tensor<1024xf32> loc(#loc133)
+    %tmp29 = arith.addf %tmp8, %tmp28 : tensor<1024xf32> loc(#loc134)
+    %9 = tt.splat %out_ptr0 : !tt.ptr<bf16> -> tensor<1024x!tt.ptr<bf16>> loc(#loc67)
+    %10 = tt.addptr %9, %xindex_5 : tensor<1024x!tt.ptr<bf16>>, tensor<1024xi32> loc(#loc67)
+    %11 = arith.truncf %tmp29 : tensor<1024xf32> to tensor<1024xbf16> loc(#loc68)
+    tt.store %10, %11, %xmask_6 : tensor<1024x!tt.ptr<bf16>> loc(#loc68)
+    tt.return loc(#loc69)
+  } loc(#loc)
+} loc(#loc)
+#loc1 = loc(unknown)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":19:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:36)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":20:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":21:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:21)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":23:28)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":24:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":25:19)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:30)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:35)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":26:75)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:30)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":27:35)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":29:18)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":30:18)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":31:32)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:28)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:44)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:37)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:54)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:52)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":32:62)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:39)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:35)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:30)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:46)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":33:86)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":34:18)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:31)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":38:18)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":39:19)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:35)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:41)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:54)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:31)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:68)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:60)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":40:119)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":41:13)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":44:20)
+#loc43 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:52)
+#loc44 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:47)
+#loc45 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:60)
+#loc46 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:41)
+#loc47 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:31)
+#loc48 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:81)
+#loc49 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:73)
+#loc50 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":47:132)
+#loc51 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":48:35)
+#loc52 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":43:35)
+#loc53 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":50:19)
+#loc54 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":51:34)
+#loc55 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:28)
+#loc56 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:46)
+#loc57 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:38)
+#loc58 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:54)
+#loc59 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":52:64)
+#loc60 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:40)
+#loc61 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:36)
+#loc62 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:31)
+#loc63 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:48)
+#loc64 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":53:88)
+#loc65 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":54:20)
+#loc66 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":55:19)
+#loc67 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:25)
+#loc68 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:37)
+#loc69 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/al/cal2r4tfyw6gic3ggqyud3nufnajx6xau2koieoitx6zg4wsiozm.py":56:4)
+#loc81 = loc("xoffset"(#loc2))
+#loc82 = loc("xoffset"(#loc3))
+#loc83 = loc("xindex"(#loc4))
+#loc84 = loc("xindex"(#loc5))
+#loc85 = loc("xmask"(#loc6))
+#loc86 = loc("x2"(#loc7))
+#loc87 = loc("x2"(#loc8))
+#loc88 = loc("x0"(#loc9))
+#loc89 = loc("x5"(#loc10))
+#loc90 = loc("tmp0"(#loc11))
+#loc91 = loc("tmp0"(#loc12))
+#loc92 = loc("tmp0"(#loc13))
+#loc93 = loc("tmp1"(#loc14))
+#loc94 = loc("tmp1"(#loc15))
+#loc95 = loc("tmp3"(#loc16))
+#loc96 = loc("tmp4"(#loc17))
+#loc97 = loc("tmp5"(#loc18))
+#loc98 = loc("tmp7"(#loc25))
+#loc99 = loc("tmp7"(#loc26))
+#loc100 = loc("tmp7"(#loc27))
+#loc101 = loc("tmp7"(#loc28))
+#loc102 = loc("tmp7"(#loc29))
+#loc103 = loc("tmp8"(#loc30))
+#loc104 = loc("tmp12"(#loc31))
+#loc105 = loc("tmp12"(#loc32))
+#loc106 = loc("tmp13"(#loc33))
+#loc107 = loc("tmp14"(#loc34))
+#loc108 = loc("tmp14"(#loc35))
+#loc109 = loc("tmp14"(#loc36))
+#loc110 = loc("tmp14"(#loc37))
+#loc111 = loc("tmp14"(#loc38))
+#loc112 = loc("tmp14"(#loc39))
+#loc113 = loc("tmp14"(#loc40))
+#loc114 = loc("tmp15"(#loc41))
+#loc115 = loc("tmp18"(#loc42))
+#loc116 = loc("tmp21"(#loc43))
+#loc117 = loc("tmp21"(#loc44))
+#loc118 = loc("tmp21"(#loc45))
+#loc119 = loc("tmp21"(#loc46))
+#loc120 = loc("tmp21"(#loc47))
+#loc121 = loc("tmp21"(#loc48))
+#loc122 = loc("tmp21"(#loc49))
+#loc123 = loc("tmp21"(#loc50))
+#loc124 = loc("tmp22"(#loc51))
+#loc125 = loc("tmp17"(#loc52))
+#loc126 = loc("tmp24"(#loc53))
+#loc127 = loc("tmp25"(#loc54))
+#loc128 = loc("tmp27"(#loc60))
+#loc129 = loc("tmp27"(#loc61))
+#loc130 = loc("tmp27"(#loc62))
+#loc131 = loc("tmp27"(#loc63))
+#loc132 = loc("tmp27"(#loc64))
+#loc133 = loc("tmp28"(#loc65))
+#loc134 = loc("tmp29"(#loc66))
+#loc135 = loc(fused[#loc124, #loc125])

SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/__grp__triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.source", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ttir", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ttgir", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.llir", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.ptx", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.cubin", "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/4BXPLEVNIV4ISF7IZIVK7CAM4LM5YGYX34PZFJY2Q7MUVWT7ZGUA/triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2.json

@@ -0,0 +1 @@
+{"hash": "e06ef592ad45788917e8ca2aaf880ce2d9dc1b17df1f92a71a87d94ada7fc9a8", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 8, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 16384, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_per_fused__to_copy_arange_bitwise_and_eq_gt_index_put_lt_new_zeros_scalar_tensor_sort_sum_unsqueeze_view_where_2"}

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/__grp__triton_red_fused__to_copy_clone_slice_sum_transpose_5.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_red_fused__to_copy_clone_slice_sum_transpose_5.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.source", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.cubin", "triton_red_fused__to_copy_clone_slice_sum_transpose_5.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.json

@@ -0,0 +1 @@
+{"hash": "e2867fc252660bf62700f11f1d82815dae162f391a3711d09922588be2264244", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 8, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 2048, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_red_fused__to_copy_clone_slice_sum_transpose_5"}

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.llir

@@ -0,0 +1,205 @@
+; ModuleID = 'LLVMDialectModule'
+source_filename = "LLVMDialectModule"
+target datalayout = "e-p3:32:32-p4:32:32-p5:32:32-p6:32:32-p7:32:32-i64:64-i128:128-v16:16-v32:32-n16:32:64"
+
+@global_smem = external addrspace(3) global [0 x i8], align 16
+
+; Function Attrs: nounwind
+define ptx_kernel void @triton_red_fused__to_copy_clone_slice_sum_transpose_5(ptr addrspace(1) %0, ptr addrspace(1) %1, i64 %2, i64 %3, i32 %4, i32 %5, ptr addrspace(1) readnone captures(none) %6, ptr addrspace(1) readnone captures(none) %7) local_unnamed_addr #0 !dbg !4 {
+  %9 = tail call i32 @llvm.nvvm.read.ptx.sreg.ctaid.x(), !dbg !7
+  %10 = shl i32 %9, 6, !dbg !8
+  %11 = tail call i32 @llvm.nvvm.read.ptx.sreg.tid.x(), !dbg !9
+  %12 = and i32 %11, 63, !dbg !9
+  %13 = or disjoint i32 %10, %12, !dbg !10
+  %14 = icmp slt i32 %13, %4, !dbg !11
+  %15 = lshr i32 %11, 6, !dbg !12
+  %16 = and i32 %15, 3, !dbg !12
+  %17 = sext i32 %13 to i64, !dbg !13
+  %.frozen = freeze i64 %2, !dbg !14
+  %18 = sdiv i64 %17, %.frozen, !dbg !14
+  %19 = mul i64 %18, %.frozen, !dbg !13
+  %.decomposed = sub i64 %17, %19, !dbg !13
+  %20 = icmp sgt i32 %5, 0, !dbg !15
+  br i1 %20, label %.lr.ph, label %._crit_edge, !dbg !15
+
+.lr.ph:                                           ; preds = %8
+  %21 = mul i64 %3, %2, !dbg !16
+  %22 = mul i64 %21, %18, !dbg !17
+  %23 = getelementptr i32, ptr addrspace(1) %0, i64 %.decomposed
+  %invariant.gep = getelementptr i32, ptr addrspace(1) %23, i64 %22, !dbg !15
+  %invariant.op = or i32 %15, 12, !dbg !15
+  %24 = insertelement <4 x i1> poison, i1 %14, i64 0, !dbg !18
+  %25 = shufflevector <4 x i1> %24, <4 x i1> poison, <4 x i32> zeroinitializer, !dbg !18
+  %26 = insertelement <4 x i32> poison, i32 %5, i64 0, !dbg !19
+  %27 = shufflevector <4 x i32> %26, <4 x i32> poison, <4 x i32> zeroinitializer, !dbg !19
+  br label %28, !dbg !15
+
+28:                                               ; preds = %.lr.ph, %28
+  %29 = phi i32 [ 0, %.lr.ph ], [ %67, %28 ]
+  %30 = phi <4 x i64> [ zeroinitializer, %.lr.ph ], [ %66, %28 ]
+  %31 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %32 = or disjoint i32 %29, %16, !dbg !21
+  %33 = or disjoint i32 %32, 4, !dbg !21
+  %34 = or disjoint i32 %32, 8, !dbg !21
+  %.reass = or disjoint i32 %29, %invariant.op
+  %35 = insertelement <4 x i32> poison, i32 %32, i64 0, !dbg !19
+  %36 = insertelement <4 x i32> %35, i32 %33, i64 1, !dbg !19
+  %37 = insertelement <4 x i32> %36, i32 %34, i64 2, !dbg !19
+  %38 = insertelement <4 x i32> %37, i32 %.reass, i64 3, !dbg !19
+  %39 = icmp slt <4 x i32> %38, %27, !dbg !19
+  %40 = sext i32 %32 to i64, !dbg !22
+  %41 = sext i32 %33 to i64, !dbg !22
+  %42 = sext i32 %34 to i64, !dbg !22
+  %43 = sext i32 %.reass to i64, !dbg !22
+  %44 = mul i64 %2, %40, !dbg !22
+  %45 = mul i64 %2, %41, !dbg !22
+  %46 = mul i64 %2, %42, !dbg !22
+  %47 = mul i64 %2, %43, !dbg !22
+  %gep = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %44, !dbg !23
+  %gep4 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %45, !dbg !23
+  %gep6 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %46, !dbg !23
+  %gep8 = getelementptr i32, ptr addrspace(1) %invariant.gep, i64 %47, !dbg !23
+  %48 = and <4 x i1> %25, %39, !dbg !18
+  %49 = extractelement <4 x i1> %48, i64 0, !dbg !20
+  %50 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep, i64 %31, i1 %49) #5, !dbg !20
+  %51 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %52 = extractelement <4 x i1> %48, i64 1, !dbg !20
+  %53 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep4, i64 %51, i1 %52) #5, !dbg !20
+  %54 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %55 = extractelement <4 x i1> %48, i64 2, !dbg !20
+  %56 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep6, i64 %54, i1 %55) #5, !dbg !20
+  %57 = tail call i64 asm sideeffect "mov.u64 $0, 0x0;\0A\09createpolicy.fractional.L2::evict_last.b64 $0, 1.0;", "=l"() #5, !dbg !20
+  %58 = extractelement <4 x i1> %48, i64 3, !dbg !20
+  %59 = tail call i32 asm sideeffect "mov.u32 $0, 0x0;\0A\09@$3 ld.global.L1::evict_last.L2::cache_hint.b32 { $0 }, [ $1 + 0 ], $2;", "=r,l,l,b"(ptr addrspace(1) %gep8, i64 %57, i1 %58) #5, !dbg !20
+  %60 = insertelement <4 x i32> poison, i32 %50, i64 0, !dbg !24
+  %61 = insertelement <4 x i32> %60, i32 %53, i64 1, !dbg !24
+  %62 = insertelement <4 x i32> %61, i32 %56, i64 2, !dbg !24
+  %63 = insertelement <4 x i32> %62, i32 %59, i64 3, !dbg !24
+  %64 = sext <4 x i32> %63 to <4 x i64>, !dbg !24
+  %65 = select <4 x i1> %48, <4 x i64> %64, <4 x i64> zeroinitializer, !dbg !25
+  %66 = add <4 x i64> %65, %30, !dbg !25
+  %67 = add i32 %29, 16, !dbg !15
+  %68 = icmp slt i32 %67, %5, !dbg !15
+  br i1 %68, label %28, label %._crit_edge.loopexit, !dbg !15
+
+._crit_edge.loopexit:                             ; preds = %28
+  %69 = tail call i64 @llvm.vector.reduce.add.v4i64(<4 x i64> %66), !dbg !26
+  br label %._crit_edge, !dbg !26
+
+._crit_edge:                                      ; preds = %._crit_edge.loopexit, %8
+  %70 = phi i64 [ 0, %8 ], [ %69, %._crit_edge.loopexit ], !dbg !26
+  %.idx = shl nuw nsw i32 %12, 5, !dbg !30
+  %71 = getelementptr i8, ptr addrspace(3) @global_smem, i32 %.idx, !dbg !30
+  %72 = getelementptr i64, ptr addrspace(3) %71, i32 %16, !dbg !30
+  %73 = insertelement <1 x i64> poison, i64 %70, i64 0, !dbg !30
+  tail call void asm sideeffect "@$2 st.shared.b64 [ $0 + 0 ], $1;", "r,l,b"(ptr addrspace(3) %72, <1 x i64> %73, i1 true) #5, !dbg !30
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !30
+  %74 = icmp samesign ult i32 %11, 256, !dbg !30
+  %75 = getelementptr i64, ptr addrspace(3) @global_smem, i32 %11, !dbg !30
+  %76 = tail call i64 asm sideeffect "@$2 ld.shared.b64 $0, [ $1 + 0 ];", "=l,r,b"(ptr addrspace(3) %75, i1 %74) #5, !dbg !30
+  %extelt.offset = lshr i64 %76, 32, !dbg !30
+  %77 = trunc nuw i64 %extelt.offset to i32, !dbg !30
+  %78 = trunc i64 %76 to i32, !dbg !30
+  %79 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %78, i32 2, i32 31), !dbg !30
+  %80 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %77, i32 2, i32 31), !dbg !30
+  %81 = insertelement <2 x i32> poison, i32 %79, i64 0, !dbg !30
+  %82 = insertelement <2 x i32> %81, i32 %80, i64 1, !dbg !30
+  %83 = bitcast <2 x i32> %82 to i64, !dbg !30
+  %84 = add i64 %76, %83, !dbg !26
+  %extelt.offset2 = lshr i64 %84, 32, !dbg !30
+  %85 = trunc nuw i64 %extelt.offset2 to i32, !dbg !30
+  %86 = trunc i64 %84 to i32, !dbg !30
+  %87 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %86, i32 1, i32 31), !dbg !30
+  %88 = tail call i32 @llvm.nvvm.shfl.sync.bfly.i32(i32 -1, i32 %85, i32 1, i32 31), !dbg !30
+  %89 = insertelement <2 x i32> poison, i32 %87, i64 0, !dbg !30
+  %90 = insertelement <2 x i32> %89, i32 %88, i64 1, !dbg !30
+  %91 = bitcast <2 x i32> %90 to i64, !dbg !30
+  %92 = add i64 %84, %91, !dbg !26
+  %93 = and i32 %11, 771, !dbg !30
+  %94 = icmp eq i32 %93, 0, !dbg !30
+  %95 = insertelement <1 x i64> poison, i64 %92, i64 0, !dbg !30
+  tail call void asm sideeffect "@$2 st.shared.b64 [ $0 + 0 ], $1;", "r,l,b"(ptr addrspace(3) %75, <1 x i64> %95, i1 %94) #5, !dbg !30
+  tail call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0), !dbg !30
+  %96 = load i64, ptr addrspace(3) %71, align 16, !dbg !30
+  %97 = trunc i64 %96 to i32, !dbg !31
+  %98 = icmp slt i64 %2, 2, !dbg !32
+  %99 = icmp sgt i64 %2, 1, !dbg !33
+  %100 = select i1 %99, i64 %2, i64 0, !dbg !34
+  %101 = zext i1 %98 to i64, !dbg !35
+  %102 = add i64 %100, %101, !dbg !36
+  %103 = mul i64 %18, %102, !dbg !37
+  %104 = getelementptr i32, ptr addrspace(1) %1, i64 %.decomposed, !dbg !38
+  %105 = getelementptr i32, ptr addrspace(1) %104, i64 %103, !dbg !38
+  %106 = and i32 %11, 192, !dbg !39
+  %107 = icmp eq i32 %106, 0, !dbg !39
+  %108 = and i1 %107, %14, !dbg !39
+  tail call void asm sideeffect "@$2 st.global.b32 [ $1 + 0 ], { $0 };", "r,l,b"(i32 %97, ptr addrspace(1) %105, i1 %108) #5, !dbg !39
+  ret void, !dbg !40
+}
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 2147483647) i32 @llvm.nvvm.read.ptx.sreg.ctaid.x() #1
+
+; Function Attrs: mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare noundef range(i32 0, 1024) i32 @llvm.nvvm.read.ptx.sreg.tid.x() #1
+
+; Function Attrs: convergent nocallback nounwind
+declare void @llvm.nvvm.barrier.cta.sync.aligned.all(i32) #2
+
+; Function Attrs: convergent nocallback nounwind memory(inaccessiblemem: readwrite)
+declare i32 @llvm.nvvm.shfl.sync.bfly.i32(i32, i32, i32, i32) #3
+
+; Function Attrs: nocallback nofree nosync nounwind speculatable willreturn memory(none)
+declare i64 @llvm.vector.reduce.add.v4i64(<4 x i64>) #4
+
+attributes #0 = { nounwind "nvvm.reqntid"="256" }
+attributes #1 = { mustprogress nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #2 = { convergent nocallback nounwind }
+attributes #3 = { convergent nocallback nounwind memory(inaccessiblemem: readwrite) }
+attributes #4 = { nocallback nofree nosync nounwind speculatable willreturn memory(none) }
+attributes #5 = { nounwind }
+
+!llvm.dbg.cu = !{!0}
+!llvm.module.flags = !{!2, !3}
+
+!0 = distinct !DICompileUnit(language: DW_LANG_C, file: !1, producer: "triton", isOptimized: true, runtimeVersion: 0, emissionKind: LineTablesOnly)
+!1 = !DIFile(filename: "cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py", directory: "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh")
+!2 = !{i32 2, !"Debug Info Version", i32 3}
+!3 = !{i32 4, !"nvvm-reflect-ftz", i32 1}
+!4 = distinct !DISubprogram(name: "triton_red_fused__to_copy_clone_slice_sum_transpose_5", linkageName: "triton_red_fused__to_copy_clone_slice_sum_transpose_5", scope: !1, file: !1, line: 18, type: !5, scopeLine: 18, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0)
+!5 = !DISubroutineType(cc: DW_CC_normal, types: !6)
+!6 = !{}
+!7 = !DILocation(line: 21, column: 28, scope: !4)
+!8 = !DILocation(line: 21, column: 33, scope: !4)
+!9 = !DILocation(line: 22, column: 44, scope: !4)
+!10 = !DILocation(line: 22, column: 23, scope: !4)
+!11 = !DILocation(line: 23, column: 21, scope: !4)
+!12 = !DILocation(line: 24, column: 37, scope: !4)
+!13 = !DILocation(line: 26, column: 19, scope: !4)
+!14 = !DILocation(line: 27, column: 19, scope: !4)
+!15 = !DILocation(line: 30, column: 40, scope: !4)
+!16 = !DILocation(line: 36, column: 54, scope: !4)
+!17 = !DILocation(line: 36, column: 58, scope: !4)
+!18 = !DILocation(line: 36, column: 73, scope: !4)
+!19 = !DILocation(line: 32, column: 29, scope: !4)
+!20 = !DILocation(line: 36, column: 63, scope: !4)
+!21 = !DILocation(line: 31, column: 31, scope: !4)
+!22 = !DILocation(line: 36, column: 43, scope: !4)
+!23 = !DILocation(line: 36, column: 34, scope: !4)
+!24 = !DILocation(line: 37, column: 23, scope: !4)
+!25 = !DILocation(line: 40, column: 48, scope: !4)
+!26 = !DILocation(line: 261, column: 15, scope: !27, inlinedAt: !29)
+!27 = distinct !DILexicalBlockFile(scope: !4, file: !28, discriminator: 0)
+!28 = !DIFile(filename: "standard.py", directory: "/workspace/specforge/lib/python3.11/site-packages/triton/language")
+!29 = !DILocation(line: 41, column: 25, scope: !4)
+!30 = !DILocation(line: 291, column: 36, scope: !27, inlinedAt: !29)
+!31 = !DILocation(line: 42, column: 19, scope: !4)
+!32 = !DILocation(line: 43, column: 49, scope: !4)
+!33 = !DILocation(line: 43, column: 75, scope: !4)
+!34 = !DILocation(line: 43, column: 66, scope: !4)
+!35 = !DILocation(line: 43, scope: !4)
+!36 = !DILocation(line: 43, column: 57, scope: !4)
+!37 = !DILocation(line: 43, column: 34, scope: !4)
+!38 = !DILocation(line: 43, column: 25, scope: !4)
+!39 = !DILocation(line: 43, column: 88, scope: !4)
+!40 = !DILocation(line: 43, column: 4, scope: !4)

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ptx

@@ -0,0 +1,527 @@
+//
+// Generated by LLVM NVPTX Back-End
+//
+
+.version 8.7
+.target sm_90a
+.address_size 64
+
+	// .globl	triton_red_fused__to_copy_clone_slice_sum_transpose_5 // -- Begin function triton_red_fused__to_copy_clone_slice_sum_transpose_5
+.extern .shared .align 16 .b8 global_smem[];
+                                        // @triton_red_fused__to_copy_clone_slice_sum_transpose_5
+.visible .entry triton_red_fused__to_copy_clone_slice_sum_transpose_5(
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_0,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_1,
+	.param .u64 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_2,
+	.param .u64 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_3,
+	.param .u32 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_4,
+	.param .u32 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_5,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_6,
+	.param .u64 .ptr .global .align 1 triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_7
+)
+.reqntid 256
+{
+	.reg .pred 	%p<24>;
+	.reg .b32 	%r<53>;
+	.reg .b64 	%rd<97>;
+	.loc	1 18 0                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:18:0
+$L__func_begin0:
+	.loc	1 18 0                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:18:0
+
+// %bb.0:
+	ld.param.b32 	%r13, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_5];
+	ld.param.b64 	%rd20, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_2];
+$L__tmp0:
+	.loc	1 21 28                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:21:28
+	mov.u32 	%r14, %ctaid.x;
+	.loc	1 21 33                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:21:33
+	shl.b32 	%r15, %r14, 6;
+	.loc	1 22 44                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:22:44
+	mov.u32 	%r1, %tid.x;
+	and.b32 	%r2, %r1, 63;
+	.loc	1 22 23                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:22:23
+	or.b32 	%r16, %r15, %r2;
+	.loc	1 26 19                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:26:19
+	cvt.s64.s32 	%rd1, %r16;
+	.loc	1 27 19                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:27:19
+	or.b64 	%rd23, %rd1, %rd20;
+	and.b64 	%rd24, %rd23, -4294967296;
+	setp.ne.b64 	%p5, %rd24, 0;
+	cvt.u32.u64 	%r51, %rd1;
+	@%p5 bra 	$L__BB0_2;
+	bra.uni 	$L__BB0_1;
+$L__BB0_2:
+	div.s64 	%rd91, %rd1, %rd20;
+	bra.uni 	$L__BB0_3;
+$L__BB0_1:
+	cvt.u32.u64 	%r17, %rd20;
+	div.u32 	%r19, %r51, %r17;
+	cvt.u64.u32 	%rd91, %r19;
+$L__BB0_3:
+	.loc	1 0 19                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:0:19
+	ld.param.b32 	%r12, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_4];
+	ld.param.b64 	%rd19, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_1];
+	bfe.u32 	%r4, %r1, 6, 2;
+	.loc	1 26 19                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:26:19
+	mul.lo.s64 	%rd26, %rd91, %rd20;
+	sub.s64 	%rd6, %rd1, %rd26;
+	.loc	1 30 40                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:30:40
+	setp.lt.s32 	%p6, %r13, 1;
+	mov.b64 	%rd96, 0;
+	shl.b64 	%rd90, %rd6, 2;
+	@%p6 bra 	$L__BB0_7;
+// %bb.4:                               // %.lr.ph
+	.loc	1 0 40                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:0:40
+	ld.param.b64 	%rd21, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_3];
+	ld.param.b64 	%rd18, [triton_red_fused__to_copy_clone_slice_sum_transpose_5_param_0];
+	shr.u32 	%r3, %r1, 6;
+	.loc	1 23 21                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:23:21
+	setp.lt.s32 	%p1, %r51, %r12;
+	.loc	1 36 54                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:54
+	mul.lo.s64 	%rd31, %rd21, %rd20;
+	.loc	1 36 58                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:58
+	mul.lo.s64 	%rd32, %rd31, %rd91;
+	add.s64 	%rd34, %rd18, %rd90;
+	.loc	1 30 40                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:30:40
+	shl.b64 	%rd35, %rd32, 2;
+	add.s64 	%rd7, %rd34, %rd35;
+	or.b32 	%r5, %r3, 12;
+	mov.b64 	%rd92, 0;
+	mov.b32 	%r52, 0;
+	mov.b64 	%rd93, %rd92;
+	mov.b64 	%rd94, %rd92;
+	mov.b64 	%rd95, %rd92;
+$L__BB0_5:                              // =>This Inner Loop Header: Depth=1
+	.loc	1 36 63                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:63
+	// begin inline asm
+	mov.u64 %rd36, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd36, 1.0;
+	// end inline asm
+	.loc	1 31 31                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:31:31
+	add.s32 	%r26, %r4, %r52;
+	add.s32 	%r27, %r26, 4;
+	add.s32 	%r28, %r26, 8;
+	.loc	1 32 29                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:32:29
+	add.s32 	%r29, %r5, %r52;
+	setp.lt.s32 	%p11, %r26, %r13;
+	setp.lt.s32 	%p12, %r27, %r13;
+	setp.lt.s32 	%p13, %r28, %r13;
+	setp.lt.s32 	%p14, %r29, %r13;
+	.loc	1 36 43                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:43
+	cvt.s64.s32 	%rd48, %r26;
+	cvt.s64.s32 	%rd49, %r27;
+	cvt.s64.s32 	%rd50, %r28;
+	cvt.s64.s32 	%rd51, %r29;
+	mul.lo.s64 	%rd52, %rd20, %rd48;
+	mul.lo.s64 	%rd53, %rd20, %rd49;
+	mul.lo.s64 	%rd54, %rd20, %rd50;
+	mul.lo.s64 	%rd55, %rd20, %rd51;
+	.loc	1 36 34                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:34
+	shl.b64 	%rd56, %rd52, 2;
+	add.s64 	%rd37, %rd7, %rd56;
+	shl.b64 	%rd57, %rd53, 2;
+	add.s64 	%rd40, %rd7, %rd57;
+	shl.b64 	%rd58, %rd54, 2;
+	add.s64 	%rd43, %rd7, %rd58;
+	shl.b64 	%rd59, %rd55, 2;
+	add.s64 	%rd46, %rd7, %rd59;
+	.loc	1 36 73                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:73
+	and.pred 	%p10, %p1, %p14;
+	and.pred 	%p9, %p1, %p13;
+	and.pred 	%p8, %p1, %p12;
+	and.pred 	%p7, %p1, %p11;
+	.loc	1 36 63                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:36:63
+	// begin inline asm
+	mov.u32 %r22, 0x0;
+	@%p7 ld.global.L1::evict_last.L2::cache_hint.b32 { %r22 }, [ %rd37 + 0 ], %rd36;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd39, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd39, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r23, 0x0;
+	@%p8 ld.global.L1::evict_last.L2::cache_hint.b32 { %r23 }, [ %rd40 + 0 ], %rd39;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd42, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd42, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r24, 0x0;
+	@%p9 ld.global.L1::evict_last.L2::cache_hint.b32 { %r24 }, [ %rd43 + 0 ], %rd42;
+	// end inline asm
+	// begin inline asm
+	mov.u64 %rd45, 0x0;
+	createpolicy.fractional.L2::evict_last.b64 %rd45, 1.0;
+	// end inline asm
+	// begin inline asm
+	mov.u32 %r25, 0x0;
+	@%p10 ld.global.L1::evict_last.L2::cache_hint.b32 { %r25 }, [ %rd46 + 0 ], %rd45;
+	// end inline asm
+	.loc	1 37 23                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:37:23
+	cvt.s64.s32 	%rd60, %r22;
+	cvt.s64.s32 	%rd61, %r23;
+	cvt.s64.s32 	%rd62, %r24;
+	cvt.s64.s32 	%rd63, %r25;
+	.loc	1 40 48                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:40:48
+	selp.b64 	%rd64, %rd63, 0, %p10;
+	selp.b64 	%rd65, %rd62, 0, %p9;
+	selp.b64 	%rd66, %rd61, 0, %p8;
+	selp.b64 	%rd67, %rd60, 0, %p7;
+	add.s64 	%rd92, %rd67, %rd92;
+	add.s64 	%rd93, %rd66, %rd93;
+	add.s64 	%rd94, %rd65, %rd94;
+	add.s64 	%rd95, %rd64, %rd95;
+	.loc	1 30 40                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:30:40
+	add.s32 	%r52, %r52, 16;
+	setp.lt.s32 	%p15, %r52, %r13;
+	@%p15 bra 	$L__BB0_5;
+// %bb.6:                               // %._crit_edge.loopexit
+$L__tmp1:
+	.loc	2 261 15                        // standard.py:261:15 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	add.s64 	%rd68, %rd92, %rd94;
+	add.s64 	%rd69, %rd93, %rd95;
+	add.s64 	%rd96, %rd68, %rd69;
+$L__tmp2:
+$L__BB0_7:                              // %._crit_edge
+	.loc	1 23 21                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:23:21
+	setp.lt.s32 	%p20, %r51, %r12;
+$L__tmp3:
+	.loc	2 291 36                        // standard.py:291:36 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	shl.b32 	%r35, %r2, 5;
+	mov.b32 	%r36, global_smem;
+	add.s32 	%r37, %r36, %r35;
+	shl.b32 	%r38, %r4, 3;
+	add.s32 	%r30, %r37, %r38;
+	mov.pred 	%p16, -1;
+	// begin inline asm
+	@%p16 st.shared.b64 [ %r30 + 0 ], %rd96;
+	// end inline asm
+	bar.sync 	0;
+	setp.lt.u32 	%p17, %r1, 256;
+	shl.b32 	%r39, %r1, 3;
+	add.s32 	%r31, %r36, %r39;
+	// begin inline asm
+	@%p17 ld.shared.b64 %rd71, [ %r31 + 0 ];
+	// end inline asm
+	mov.b64 	{_, %r40}, %rd71;
+	cvt.u32.u64 	%r41, %rd71;
+	shfl.sync.bfly.b32 	%r42, %r41, 2, 31, -1;
+	shfl.sync.bfly.b32 	%r43, %r40, 2, 31, -1;
+	cvt.u64.u32 	%rd74, %r42;
+	cvt.u64.u32 	%rd75, %r43;
+	shl.b64 	%rd76, %rd75, 32;
+	or.b64 	%rd77, %rd74, %rd76;
+	.loc	2 261 15                        // standard.py:261:15 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	add.s64 	%rd78, %rd71, %rd77;
+	.loc	2 291 36                        // standard.py:291:36 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	mov.b64 	{_, %r44}, %rd78;
+	cvt.u32.u64 	%r45, %rd78;
+	shfl.sync.bfly.b32 	%r46, %r45, 1, 31, -1;
+	shfl.sync.bfly.b32 	%r47, %r44, 1, 31, -1;
+	cvt.u64.u32 	%rd79, %r46;
+	cvt.u64.u32 	%rd80, %r47;
+	shl.b64 	%rd81, %rd80, 32;
+	or.b64 	%rd82, %rd79, %rd81;
+	.loc	2 261 15                        // standard.py:261:15 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	add.s64 	%rd72, %rd78, %rd82;
+	.loc	2 291 36                        // standard.py:291:36 @[ cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:41:25 ]
+	and.b32 	%r48, %r1, 771;
+	setp.eq.b32 	%p18, %r48, 0;
+	// begin inline asm
+	@%p18 st.shared.b64 [ %r31 + 0 ], %rd72;
+	// end inline asm
+	bar.sync 	0;
+	ld.shared.b32 	%r33, [%r37];
+$L__tmp4:
+	.loc	1 43 49                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:49
+	setp.lt.s64 	%p21, %rd20, 2;
+	.loc	1 43 75                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:75
+	setp.gt.s64 	%p22, %rd20, 1;
+	.loc	1 43 66                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:66
+	selp.b64 	%rd83, %rd20, 0, %p22;
+	.loc	1 43 0                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43
+	selp.b64 	%rd84, 1, 0, %p21;
+	.loc	1 43 57                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:57
+	add.s64 	%rd85, %rd83, %rd84;
+	.loc	1 43 34                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:34
+	mul.lo.s64 	%rd86, %rd91, %rd85;
+	.loc	1 43 25                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:25
+	add.s64 	%rd88, %rd19, %rd90;
+	shl.b64 	%rd89, %rd86, 2;
+	add.s64 	%rd73, %rd88, %rd89;
+	.loc	1 43 88                         // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:88
+	and.b32 	%r49, %r1, 192;
+	setp.eq.b32 	%p23, %r49, 0;
+	and.pred 	%p19, %p23, %p20;
+	// begin inline asm
+	@%p19 st.global.b32 [ %rd73 + 0 ], { %r33 };
+	// end inline asm
+	.loc	1 43 4                          // cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py:43:4
+	ret;
+$L__tmp5:
+$L__func_end0:
+                                        // -- End function
+}
+	.file	1 "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py"
+	.file	2 "/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py"
+	.section	.debug_abbrev
+	{
+.b8 1                                   // Abbreviation Code
+.b8 17                                  // DW_TAG_compile_unit
+.b8 1                                   // DW_CHILDREN_yes
+.b8 37                                  // DW_AT_producer
+.b8 8                                   // DW_FORM_string
+.b8 19                                  // DW_AT_language
+.b8 5                                   // DW_FORM_data2
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 16                                  // DW_AT_stmt_list
+.b8 6                                   // DW_FORM_data4
+.b8 27                                  // DW_AT_comp_dir
+.b8 8                                   // DW_FORM_string
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 2                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 0                                   // DW_CHILDREN_no
+.b8 3                                   // DW_AT_name
+.b8 8                                   // DW_FORM_string
+.b8 32                                  // DW_AT_inline
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 3                                   // Abbreviation Code
+.b8 46                                  // DW_TAG_subprogram
+.b8 1                                   // DW_CHILDREN_yes
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 4                                   // Abbreviation Code
+.b8 29                                  // DW_TAG_inlined_subroutine
+.b8 0                                   // DW_CHILDREN_no
+.b8 49                                  // DW_AT_abstract_origin
+.b8 19                                  // DW_FORM_ref4
+.b8 17                                  // DW_AT_low_pc
+.b8 1                                   // DW_FORM_addr
+.b8 18                                  // DW_AT_high_pc
+.b8 1                                   // DW_FORM_addr
+.b8 88                                  // DW_AT_call_file
+.b8 11                                  // DW_FORM_data1
+.b8 89                                  // DW_AT_call_line
+.b8 11                                  // DW_FORM_data1
+.b8 87                                  // DW_AT_call_column
+.b8 11                                  // DW_FORM_data1
+.b8 0                                   // EOM(1)
+.b8 0                                   // EOM(2)
+.b8 0                                   // EOM(3)
+	}
+	.section	.debug_info
+	{
+.b32 238                                // Length of Unit
+.b8 2                                   // DWARF version number
+.b8 0
+.b32 .debug_abbrev                      // Offset Into Abbrev. Section
+.b8 8                                   // Address Size (in bytes)
+.b8 1                                   // Abbrev [1] 0xb:0xe7 DW_TAG_compile_unit
+.b8 116                                 // DW_AT_producer
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 0
+.b8 2                                   // DW_AT_language
+.b8 0
+.b8 99                                  // DW_AT_name
+.b8 119
+.b8 104
+.b8 103
+.b8 117
+.b8 50
+.b8 98
+.b8 122
+.b8 102
+.b8 107
+.b8 115
+.b8 101
+.b8 99
+.b8 98
+.b8 109
+.b8 52
+.b8 105
+.b8 110
+.b8 116
+.b8 51
+.b8 111
+.b8 107
+.b8 107
+.b8 104
+.b8 51
+.b8 112
+.b8 104
+.b8 53
+.b8 52
+.b8 107
+.b8 100
+.b8 109
+.b8 115
+.b8 54
+.b8 122
+.b8 102
+.b8 97
+.b8 100
+.b8 55
+.b8 110
+.b8 103
+.b8 112
+.b8 102
+.b8 122
+.b8 113
+.b8 54
+.b8 104
+.b8 50
+.b8 111
+.b8 119
+.b8 119
+.b8 119
+.b8 46
+.b8 112
+.b8 121
+.b8 0
+.b32 .debug_line                        // DW_AT_stmt_list
+.b8 47                                  // DW_AT_comp_dir
+.b8 119
+.b8 111
+.b8 114
+.b8 107
+.b8 115
+.b8 112
+.b8 97
+.b8 99
+.b8 101
+.b8 47
+.b8 104
+.b8 97
+.b8 110
+.b8 114
+.b8 117
+.b8 105
+.b8 47
+.b8 83
+.b8 112
+.b8 101
+.b8 99
+.b8 70
+.b8 111
+.b8 114
+.b8 103
+.b8 101
+.b8 45
+.b8 101
+.b8 120
+.b8 116
+.b8 47
+.b8 99
+.b8 97
+.b8 99
+.b8 104
+.b8 101
+.b8 47
+.b8 99
+.b8 111
+.b8 109
+.b8 112
+.b8 105
+.b8 108
+.b8 101
+.b8 100
+.b8 95
+.b8 107
+.b8 101
+.b8 114
+.b8 110
+.b8 101
+.b8 108
+.b8 115
+.b8 47
+.b8 119
+.b8 104
+.b8 0
+.b8 2                                   // Abbrev [2] 0x8b:0x38 DW_TAG_subprogram
+.b8 116                                 // DW_AT_name
+.b8 114
+.b8 105
+.b8 116
+.b8 111
+.b8 110
+.b8 95
+.b8 114
+.b8 101
+.b8 100
+.b8 95
+.b8 102
+.b8 117
+.b8 115
+.b8 101
+.b8 100
+.b8 95
+.b8 95
+.b8 116
+.b8 111
+.b8 95
+.b8 99
+.b8 111
+.b8 112
+.b8 121
+.b8 95
+.b8 99
+.b8 108
+.b8 111
+.b8 110
+.b8 101
+.b8 95
+.b8 115
+.b8 108
+.b8 105
+.b8 99
+.b8 101
+.b8 95
+.b8 115
+.b8 117
+.b8 109
+.b8 95
+.b8 116
+.b8 114
+.b8 97
+.b8 110
+.b8 115
+.b8 112
+.b8 111
+.b8 115
+.b8 101
+.b8 95
+.b8 53
+.b8 0
+.b8 1                                   // DW_AT_inline
+.b8 3                                   // Abbrev [3] 0xc3:0x2e DW_TAG_subprogram
+.b64 $L__func_begin0                    // DW_AT_low_pc
+.b64 $L__func_end0                      // DW_AT_high_pc
+.b32 139                                // DW_AT_abstract_origin
+.b8 4                                   // Abbrev [4] 0xd8:0x18 DW_TAG_inlined_subroutine
+.b32 139                                // DW_AT_abstract_origin
+.b64 $L__tmp1                           // DW_AT_low_pc
+.b64 $L__tmp4                           // DW_AT_high_pc
+.b8 1                                   // DW_AT_call_file
+.b8 41                                  // DW_AT_call_line
+.b8 25                                  // DW_AT_call_column
+.b8 0                                   // End Of Children Mark
+.b8 0                                   // End Of Children Mark
+	}
+	.section	.debug_macinfo	{	}

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.source

@@ -0,0 +1,193 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":18:0)
+#loc41 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":285:0)
+#loc43 = loc(unknown)
+#loc46 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":260:0)
+#loc50 = loc("in_ptr0"(#loc))
+#loc51 = loc("out_ptr1"(#loc))
+#loc52 = loc("ks0"(#loc))
+#loc53 = loc("ks1"(#loc))
+#loc54 = loc("xnumel"(#loc))
+#loc55 = loc("r0_numel"(#loc))
+#loc85 = loc("input"(#loc41))
+#loc86 = loc("a"(#loc46))
+#loc87 = loc("b"(#loc46))
+module {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %xoffset = tt.get_program_id x : i32 loc(#loc56)
+    %xoffset_0 = arith.constant 64 : i32 loc(#loc57)
+    %xoffset_1 = arith.constant 64 : i32 loc(#loc57)
+    %xoffset_2 = arith.muli %xoffset, %xoffset_1 : i32 loc(#loc57)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32> loc(#loc58)
+    %xindex_3 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc59)
+    %xindex_4 = tt.splat %xoffset_2 : i32 -> tensor<64x1xi32> loc(#loc60)
+    %xindex_5 = arith.addi %xindex_4, %xindex_3 : tensor<64x1xi32> loc(#loc60)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32> loc(#loc61)
+    %xmask_6 = arith.cmpi slt, %xindex_5, %xmask : tensor<64x1xi32> loc(#loc61)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32> loc(#loc62)
+    %r0_base_7 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32> -> tensor<1x16xi32> loc(#loc63)
+    %x0 = arith.extsi %xindex_5 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc64)
+    %x0_8 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc64)
+    %x0_9 = arith.remsi %x0, %x0_8 : tensor<64x1xi64> loc(#loc64)
+    %x1 = arith.extsi %xindex_5 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc65)
+    %x1_10 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc65)
+    %x1_11 = arith.divsi %x1, %x1_10 : tensor<64x1xi64> loc(#loc65)
+    %_tmp3 = arith.constant 0 : i64 loc(#loc66)
+    %_tmp3_12 = arith.constant dense<0> : tensor<64x16xi64> loc(#loc66)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc12)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc12)
+    %0 = arith.bitcast %c0_i32 : i32 to i32 loc(#loc12)
+    %1 = arith.bitcast %r0_numel : i32 to i32 loc(#loc12)
+    %2 = arith.bitcast %c16_i32 : i32 to i32 loc(#loc12)
+    %3 = ub.poison : i32 loc(#loc12)
+    %_tmp3_13 = scf.for %r0_offset = %0 to %1 step %2 iter_args(%_tmp3_18 = %_tmp3_12) -> (tensor<64x16xi64>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x16xi32> loc(#loc68)
+      %r0_index_19 = arith.addi %r0_index, %r0_base_7 : tensor<1x16xi32> loc(#loc68)
+      %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32> loc(#loc69)
+      %r0_mask_20 = arith.cmpi slt, %r0_index_19, %r0_mask : tensor<1x16xi32> loc(#loc69)
+      %tmp0 = arith.extsi %r0_index_19 : tensor<1x16xi32> to tensor<1x16xi64> loc(#loc70)
+      %tmp0_21 = tt.splat %ks0 : i64 -> tensor<1x16xi64> loc(#loc70)
+      %tmp0_22 = arith.muli %tmp0_21, %tmp0 : tensor<1x16xi64> loc(#loc70)
+      %tmp0_23 = tt.broadcast %x0_9 : tensor<64x1xi64> -> tensor<64x16xi64> loc(#loc71)
+      %tmp0_24 = tt.broadcast %tmp0_22 : tensor<1x16xi64> -> tensor<64x16xi64> loc(#loc71)
+      %tmp0_25 = arith.addi %tmp0_23, %tmp0_24 : tensor<64x16xi64> loc(#loc71)
+      %tmp0_26 = arith.muli %ks0, %ks1 : i64 loc(#loc72)
+      %tmp0_27 = tt.splat %tmp0_26 : i64 -> tensor<64x1xi64> loc(#loc73)
+      %tmp0_28 = arith.muli %tmp0_27, %x1_11 : tensor<64x1xi64> loc(#loc73)
+      %tmp0_29 = tt.broadcast %tmp0_28 : tensor<64x1xi64> -> tensor<64x16xi64> loc(#loc74)
+      %tmp0_30 = arith.addi %tmp0_25, %tmp0_29 : tensor<64x16xi64> loc(#loc74)
+      %tmp0_31 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<64x16x!tt.ptr<i32>> loc(#loc75)
+      %tmp0_32 = tt.addptr %tmp0_31, %tmp0_30 : tensor<64x16x!tt.ptr<i32>>, tensor<64x16xi64> loc(#loc75)
+      %tmp0_33 = tt.broadcast %r0_mask_20 : tensor<1x16xi1> -> tensor<64x16xi1> loc(#loc76)
+      %tmp0_34 = tt.broadcast %xmask_6 : tensor<64x1xi1> -> tensor<64x16xi1> loc(#loc76)
+      %tmp0_35 = arith.andi %tmp0_33, %tmp0_34 : tensor<64x16xi1> loc(#loc76)
+      %tmp0_36 = arith.constant 0.000000e+00 : f32 loc(#loc77)
+      %tmp0_37 = arith.constant dense<0.000000e+00> : tensor<64x16xf32> loc(#loc77)
+      %tmp0_38 = arith.fptosi %tmp0_37 : tensor<64x16xf32> to tensor<64x16xi32> loc(#loc77)
+      %tmp0_39 = tt.load %tmp0_32, %tmp0_35, %tmp0_38 evictionPolicy = evict_last : tensor<64x16x!tt.ptr<i32>> loc(#loc77)
+      %tmp1 = arith.extsi %tmp0_39 : tensor<64x16xi32> to tensor<64x16xi64> loc(#loc78)
+      %tmp4 = arith.addi %_tmp3_18, %tmp1 : tensor<64x16xi64> loc(#loc79)
+      %_tmp3_40 = tt.broadcast %r0_mask_20 : tensor<1x16xi1> -> tensor<64x16xi1> loc(#loc80)
+      %_tmp3_41 = tt.broadcast %xmask_6 : tensor<64x1xi1> -> tensor<64x16xi1> loc(#loc80)
+      %_tmp3_42 = arith.andi %_tmp3_40, %_tmp3_41 : tensor<64x16xi1> loc(#loc80)
+      %_tmp3_43 = arith.select %_tmp3_42, %tmp4, %_tmp3_18 : tensor<64x16xi1>, tensor<64x16xi64> loc(#loc81)
+      scf.yield %_tmp3_43 : tensor<64x16xi64> loc(#loc27)
+    } loc(#loc67)
+    %tmp3 = tt.call @"triton.language.standard.sum__i64S64_16S__(1,)cconstexpr_1__(2,)cconstexpr_False__(3,)cNone"(%_tmp3_13) : (tensor<64x16xi64>) -> tensor<64xi64> loc(#loc82)
+    %tmp3_14 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<64xi64> -> tensor<64x1xi64> loc(#loc83)
+    %tmp5 = arith.trunci %tmp3_14 : tensor<64x1xi64> to tensor<64x1xi32> loc(#loc84)
+    %c1_i32 = arith.constant 1 : i32 loc(#loc31)
+    %4 = arith.extsi %c1_i32 : i32 to i64 loc(#loc31)
+    %5 = arith.cmpi sge, %4, %ks0 : i64 loc(#loc31)
+    %c1_i32_15 = arith.constant 1 : i32 loc(#loc32)
+    %c1_i32_16 = arith.constant 1 : i32 loc(#loc32)
+    %6 = arith.extui %5 : i1 to i32 loc(#loc32)
+    %7 = arith.muli %c1_i32_16, %6 : i32 loc(#loc32)
+    %c1_i32_17 = arith.constant 1 : i32 loc(#loc33)
+    %8 = arith.extsi %c1_i32_17 : i32 to i64 loc(#loc33)
+    %9 = arith.cmpi sgt, %ks0, %8 : i64 loc(#loc33)
+    %10 = arith.extui %9 : i1 to i64 loc(#loc34)
+    %11 = arith.muli %ks0, %10 : i64 loc(#loc34)
+    %12 = arith.extsi %7 : i32 to i64 loc(#loc35)
+    %13 = arith.addi %12, %11 : i64 loc(#loc35)
+    %14 = tt.splat %13 : i64 -> tensor<64x1xi64> loc(#loc36)
+    %15 = arith.muli %x1_11, %14 : tensor<64x1xi64> loc(#loc36)
+    %16 = arith.addi %x0_9, %15 : tensor<64x1xi64> loc(#loc37)
+    %17 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<64x1x!tt.ptr<i32>> loc(#loc38)
+    %18 = tt.addptr %17, %16 : tensor<64x1x!tt.ptr<i32>>, tensor<64x1xi64> loc(#loc38)
+    tt.store %18, %tmp5, %xmask_6 : tensor<64x1x!tt.ptr<i32>> loc(#loc39)
+    tt.return loc(#loc40)
+  } loc(#loc)
+  tt.func private @"triton.language.standard.sum__i64S64_16S__(1,)cconstexpr_1__(2,)cconstexpr_False__(3,)cNone"(%input: tensor<64x16xi64> loc("input"(#loc41))) -> tensor<64xi64> attributes {noinline = false} {
+    %0 = "tt.reduce"(%input) <{axis = 1 : i32}> ({
+    ^bb0(%arg1: i64 loc(unknown), %arg2: i64 loc(unknown)):
+      %2 = tt.call @triton.language.standard._sum_combine__i64_i64__(%arg1, %arg2) : (i64, i64) -> i64 loc(#loc42)
+      tt.reduce.return %2 : i64 loc(#loc42)
+    }) : (tensor<64x16xi64>) -> tensor<64xi64> loc(#loc42)
+    tt.return %0 : tensor<64xi64> loc(#loc44)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : tensor<64xi64> loc(#loc45)
+    tt.return %1 : tensor<64xi64> loc(#loc45)
+  } loc(#loc41)
+  tt.func private @triton.language.standard._sum_combine__i64_i64__(%a: i64 loc("a"(#loc46)), %b: i64 loc("b"(#loc46))) -> i64 attributes {noinline = false} {
+    %0 = arith.addi %a, %b : i64 loc(#loc47)
+    tt.return %0 : i64 loc(#loc48)
+  ^bb1:  // no predecessors
+    %1 = ub.poison : i64 loc(#loc49)
+    tt.return %1 : i64 loc(#loc49)
+  } loc(#loc46)
+} loc(#loc)
+#loc1 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:28)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:33)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:36)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":23:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":24:27)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":24:37)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":26:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":27:19)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":28:43)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":30:40)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":31:31)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":32:29)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:43)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:39)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:54)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:58)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:50)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:34)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:73)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:63)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":37:23)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":39:23)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:35)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:48)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:8)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:25)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:28)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":42:19)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:49)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:41)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:75)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:66)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:57)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:34)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:30)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:25)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:88)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:4)
+#loc42 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc44 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:11)
+#loc45 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:4)
+#loc47 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc48 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:11)
+#loc49 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:4)
+#loc56 = loc("xoffset"(#loc1))
+#loc57 = loc("xoffset"(#loc2))
+#loc58 = loc("xindex"(#loc3))
+#loc59 = loc("xindex"(#loc4))
+#loc60 = loc("xindex"(#loc5))
+#loc61 = loc("xmask"(#loc6))
+#loc62 = loc("r0_base"(#loc7))
+#loc63 = loc("r0_base"(#loc8))
+#loc64 = loc("x0"(#loc9))
+#loc65 = loc("x1"(#loc10))
+#loc66 = loc("_tmp3"(#loc11))
+#loc67 = loc("_tmp3"(#loc12))
+#loc68 = loc("r0_index"(#loc13))
+#loc69 = loc("r0_mask"(#loc14))
+#loc70 = loc("tmp0"(#loc15))
+#loc71 = loc("tmp0"(#loc16))
+#loc72 = loc("tmp0"(#loc17))
+#loc73 = loc("tmp0"(#loc18))
+#loc74 = loc("tmp0"(#loc19))
+#loc75 = loc("tmp0"(#loc20))
+#loc76 = loc("tmp0"(#loc21))
+#loc77 = loc("tmp0"(#loc22))
+#loc78 = loc("tmp1"(#loc23))
+#loc79 = loc("tmp4"(#loc24))
+#loc80 = loc("_tmp3"(#loc25))
+#loc81 = loc("_tmp3"(#loc26))
+#loc82 = loc("tmp3"(#loc28))
+#loc83 = loc("tmp3"(#loc29))
+#loc84 = loc("tmp5"(#loc30))

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttgir

@@ -0,0 +1,147 @@
+#blocked = #ttg.blocked<{sizePerThread = [1, 1], threadsPerWarp = [32, 1], warpsPerCTA = [2, 4], order = [0, 1]}>
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":18:0)
+#loc1 = loc(unknown)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:25)
+#loc40 = loc("in_ptr0"(#loc))
+#loc41 = loc("out_ptr1"(#loc))
+#loc42 = loc("ks0"(#loc))
+#loc43 = loc("ks1"(#loc))
+#loc44 = loc("xnumel"(#loc))
+#loc45 = loc("r0_numel"(#loc))
+#loc68 = loc("tmp3"(#loc26))
+#loc73 = loc(callsite(#loc1 at #loc68))
+module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 8 : i32, ttg.target = "cuda:90", "ttg.threads-per-warp" = 32 : i32} {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<0> : tensor<64x16xi64, #blocked> loc(#loc1)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc1)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc1)
+    %c1_i64 = arith.constant 1 : i64 loc(#loc1)
+    %c64_i32 = arith.constant 64 : i32 loc(#loc1)
+    %cst_0 = arith.constant dense<0> : tensor<64x16xi32, #blocked> loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc46)
+    %xoffset_1 = arith.muli %xoffset, %c64_i32 : i32 loc(#loc47)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc48)
+    %xindex_2 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<64x1xi32, #blocked> loc(#loc48)
+    %xindex_3 = tt.splat %xoffset_1 : i32 -> tensor<64x1xi32, #blocked> loc(#loc49)
+    %xindex_4 = arith.addi %xindex_3, %xindex_2 : tensor<64x1xi32, #blocked> loc(#loc49)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32, #blocked> loc(#loc50)
+    %xmask_5 = arith.cmpi slt, %xindex_4, %xmask : tensor<64x1xi32, #blocked> loc(#loc50)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> loc(#loc51)
+    %r0_base_6 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> -> tensor<1x16xi32, #blocked> loc(#loc51)
+    %x0 = arith.extsi %xindex_4 : tensor<64x1xi32, #blocked> to tensor<64x1xi64, #blocked> loc(#loc52)
+    %x0_7 = tt.splat %ks0 : i64 -> tensor<64x1xi64, #blocked> loc(#loc52)
+    %x0_8 = arith.remsi %x0, %x0_7 : tensor<64x1xi64, #blocked> loc(#loc52)
+    %x1 = arith.divsi %x0, %x0_7 : tensor<64x1xi64, #blocked> loc(#loc53)
+    %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32, #blocked> loc(#loc54)
+    %tmp0 = tt.splat %ks0 : i64 -> tensor<1x16xi64, #blocked> loc(#loc55)
+    %tmp0_9 = tt.broadcast %x0_8 : tensor<64x1xi64, #blocked> -> tensor<64x16xi64, #blocked> loc(#loc56)
+    %tmp0_10 = arith.muli %ks0, %ks1 : i64 loc(#loc57)
+    %tmp0_11 = tt.splat %tmp0_10 : i64 -> tensor<64x1xi64, #blocked> loc(#loc58)
+    %tmp0_12 = arith.muli %tmp0_11, %x1 : tensor<64x1xi64, #blocked> loc(#loc58)
+    %tmp0_13 = tt.broadcast %tmp0_12 : tensor<64x1xi64, #blocked> -> tensor<64x16xi64, #blocked> loc(#loc59)
+    %tmp0_14 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<64x16x!tt.ptr<i32>, #blocked> loc(#loc60)
+    %tmp0_15 = tt.broadcast %xmask_5 : tensor<64x1xi1, #blocked> -> tensor<64x16xi1, #blocked> loc(#loc61)
+    %_tmp3 = scf.for %_tmp3_17 = %c0_i32 to %r0_numel step %c16_i32 iter_args(%_tmp3_18 = %cst) -> (tensor<64x16xi64, #blocked>)  : i32 {
+      %r0_index = tt.splat %_tmp3_17 : i32 -> tensor<1x16xi32, #blocked> loc(#loc63)
+      %r0_index_19 = arith.addi %r0_index, %r0_base_6 : tensor<1x16xi32, #blocked> loc(#loc63)
+      %r0_mask_20 = arith.cmpi slt, %r0_index_19, %r0_mask : tensor<1x16xi32, #blocked> loc(#loc54)
+      %tmp0_21 = arith.extsi %r0_index_19 : tensor<1x16xi32, #blocked> to tensor<1x16xi64, #blocked> loc(#loc55)
+      %tmp0_22 = arith.muli %tmp0, %tmp0_21 : tensor<1x16xi64, #blocked> loc(#loc55)
+      %tmp0_23 = tt.broadcast %tmp0_22 : tensor<1x16xi64, #blocked> -> tensor<64x16xi64, #blocked> loc(#loc56)
+      %tmp0_24 = arith.addi %tmp0_9, %tmp0_23 : tensor<64x16xi64, #blocked> loc(#loc56)
+      %tmp0_25 = arith.addi %tmp0_24, %tmp0_13 : tensor<64x16xi64, #blocked> loc(#loc59)
+      %tmp0_26 = tt.addptr %tmp0_14, %tmp0_25 : tensor<64x16x!tt.ptr<i32>, #blocked>, tensor<64x16xi64, #blocked> loc(#loc60)
+      %tmp0_27 = tt.broadcast %r0_mask_20 : tensor<1x16xi1, #blocked> -> tensor<64x16xi1, #blocked> loc(#loc61)
+      %tmp0_28 = arith.andi %tmp0_27, %tmp0_15 : tensor<64x16xi1, #blocked> loc(#loc61)
+      %tmp0_29 = tt.load %tmp0_26, %tmp0_28, %cst_0 evictionPolicy = evict_last : tensor<64x16x!tt.ptr<i32>, #blocked> loc(#loc64)
+      %tmp1 = arith.extsi %tmp0_29 : tensor<64x16xi32, #blocked> to tensor<64x16xi64, #blocked> loc(#loc65)
+      %tmp4 = arith.addi %_tmp3_18, %tmp1 : tensor<64x16xi64, #blocked> loc(#loc66)
+      %_tmp3_30 = arith.select %tmp0_28, %tmp4, %_tmp3_18 : tensor<64x16xi1, #blocked>, tensor<64x16xi64, #blocked> loc(#loc67)
+      scf.yield %_tmp3_30 : tensor<64x16xi64, #blocked> loc(#loc24)
+    } loc(#loc62)
+    %tmp3 = "tt.reduce"(%_tmp3) <{axis = 1 : i32}> ({
+    ^bb0(%tmp3_17: i64 loc(callsite(#loc1 at #loc68)), %tmp3_18: i64 loc(callsite(#loc1 at #loc68))):
+      %tmp3_19 = arith.addi %tmp3_17, %tmp3_18 : i64 loc(#loc74)
+      tt.reduce.return %tmp3_19 : i64 loc(#loc72)
+    }) : (tensor<64x16xi64, #blocked>) -> tensor<64xi64, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc72)
+    %tmp3_16 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<64xi64, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<64x1xi64, #blocked> loc(#loc69)
+    %tmp5 = arith.trunci %tmp3_16 : tensor<64x1xi64, #blocked> to tensor<64x1xi32, #blocked> loc(#loc70)
+    %0 = arith.cmpi sle, %ks0, %c1_i64 : i64 loc(#loc30)
+    %1 = arith.cmpi sgt, %ks0, %c1_i64 : i64 loc(#loc31)
+    %2 = arith.extui %1 : i1 to i64 loc(#loc32)
+    %3 = arith.muli %ks0, %2 : i64 loc(#loc32)
+    %4 = arith.extui %0 : i1 to i64 loc(#loc71)
+    %5 = arith.addi %4, %3 : i64 loc(#loc33)
+    %6 = tt.splat %5 : i64 -> tensor<64x1xi64, #blocked> loc(#loc35)
+    %7 = arith.muli %x1, %6 : tensor<64x1xi64, #blocked> loc(#loc35)
+    %8 = arith.addi %x0_8, %7 : tensor<64x1xi64, #blocked> loc(#loc36)
+    %9 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<64x1x!tt.ptr<i32>, #blocked> loc(#loc37)
+    %10 = tt.addptr %9, %8 : tensor<64x1x!tt.ptr<i32>, #blocked>, tensor<64x1xi64, #blocked> loc(#loc37)
+    tt.store %10, %tmp5, %xmask_5 : tensor<64x1x!tt.ptr<i32>, #blocked> loc(#loc38)
+    tt.return loc(#loc39)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":23:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":24:37)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":26:19)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":27:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":32:29)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:43)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:39)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:54)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:58)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:50)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:34)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:73)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":30:40)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":31:31)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:63)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":37:23)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":39:23)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:48)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:8)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc28 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:28)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":42:19)
+#loc30 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:49)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:75)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:66)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:57)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:41)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:34)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:30)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:25)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:88)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:4)
+#loc46 = loc("xoffset"(#loc2))
+#loc47 = loc("xoffset"(#loc3))
+#loc48 = loc("xindex"(#loc4))
+#loc49 = loc("xindex"(#loc5))
+#loc50 = loc("xmask"(#loc6))
+#loc51 = loc("r0_base"(#loc7))
+#loc52 = loc("x0"(#loc8))
+#loc53 = loc("x1"(#loc9))
+#loc54 = loc("r0_mask"(#loc10))
+#loc55 = loc("tmp0"(#loc11))
+#loc56 = loc("tmp0"(#loc12))
+#loc57 = loc("tmp0"(#loc13))
+#loc58 = loc("tmp0"(#loc14))
+#loc59 = loc("tmp0"(#loc15))
+#loc60 = loc("tmp0"(#loc16))
+#loc61 = loc("tmp0"(#loc17))
+#loc62 = loc("_tmp3"(#loc18))
+#loc63 = loc("r0_index"(#loc19))
+#loc64 = loc("tmp0"(#loc20))
+#loc65 = loc("tmp1"(#loc21))
+#loc66 = loc("tmp4"(#loc22))
+#loc67 = loc("_tmp3"(#loc23))
+#loc69 = loc("tmp3"(#loc28))
+#loc70 = loc("tmp5"(#loc29))
+#loc71 = loc(fused[#loc33, #loc34])
+#loc72 = loc(callsite(#loc25 at #loc68))
+#loc74 = loc(callsite(#loc27 at #loc72))

SpecForge-ext/cache/compiled_kernels/triton/0/4KDH7QSSMYF7MJYA6EPR3AUBLWXBMLZZDI3RDUEZEJMIXYRGIJCA/triton_red_fused__to_copy_clone_slice_sum_transpose_5.ttir

@@ -0,0 +1,152 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":18:0)
+#loc1 = loc(unknown)
+#loc29 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:25)
+#loc43 = loc("in_ptr0"(#loc))
+#loc44 = loc("out_ptr1"(#loc))
+#loc45 = loc("ks0"(#loc))
+#loc46 = loc("ks1"(#loc))
+#loc47 = loc("xnumel"(#loc))
+#loc48 = loc("r0_numel"(#loc))
+#loc74 = loc("tmp3"(#loc29))
+#loc79 = loc(callsite(#loc1 at #loc74))
+module {
+  tt.func public @triton_red_fused__to_copy_clone_slice_sum_transpose_5(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr1: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr1"(#loc)), %ks0: i64 loc("ks0"(#loc)), %ks1: i64 loc("ks1"(#loc)), %xnumel: i32 loc("xnumel"(#loc)), %r0_numel: i32 loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %c1_i64 = arith.constant 1 : i64 loc(#loc1)
+    %cst = arith.constant dense<0> : tensor<64x16xi32> loc(#loc1)
+    %c16_i32 = arith.constant 16 : i32 loc(#loc2)
+    %c0_i32 = arith.constant 0 : i32 loc(#loc2)
+    %_tmp3 = arith.constant dense<0> : tensor<64x16xi64> loc(#loc49)
+    %c64_i32 = arith.constant 64 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc50)
+    %xoffset_0 = arith.muli %xoffset, %c64_i32 : i32 loc(#loc51)
+    %xindex = tt.make_range {end = 64 : i32, start = 0 : i32} : tensor<64xi32> loc(#loc52)
+    %xindex_1 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<64xi32> -> tensor<64x1xi32> loc(#loc53)
+    %xindex_2 = tt.splat %xoffset_0 : i32 -> tensor<64x1xi32> loc(#loc54)
+    %xindex_3 = arith.addi %xindex_2, %xindex_1 : tensor<64x1xi32> loc(#loc54)
+    %xmask = tt.splat %xnumel : i32 -> tensor<64x1xi32> loc(#loc55)
+    %xmask_4 = arith.cmpi slt, %xindex_3, %xmask : tensor<64x1xi32> loc(#loc55)
+    %r0_base = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32> loc(#loc56)
+    %r0_base_5 = tt.expand_dims %r0_base {axis = 0 : i32} : tensor<16xi32> -> tensor<1x16xi32> loc(#loc57)
+    %x0 = arith.extsi %xindex_3 : tensor<64x1xi32> to tensor<64x1xi64> loc(#loc58)
+    %x0_6 = tt.splat %ks0 : i64 -> tensor<64x1xi64> loc(#loc58)
+    %x0_7 = arith.remsi %x0, %x0_6 : tensor<64x1xi64> loc(#loc58)
+    %x1 = arith.divsi %x0, %x0_6 : tensor<64x1xi64> loc(#loc59)
+    %_tmp3_8 = scf.for %r0_offset = %c0_i32 to %r0_numel step %c16_i32 iter_args(%_tmp3_10 = %_tmp3) -> (tensor<64x16xi64>)  : i32 {
+      %r0_index = tt.splat %r0_offset : i32 -> tensor<1x16xi32> loc(#loc61)
+      %r0_index_11 = arith.addi %r0_index, %r0_base_5 : tensor<1x16xi32> loc(#loc61)
+      %r0_mask = tt.splat %r0_numel : i32 -> tensor<1x16xi32> loc(#loc62)
+      %r0_mask_12 = arith.cmpi slt, %r0_index_11, %r0_mask : tensor<1x16xi32> loc(#loc62)
+      %tmp0 = arith.extsi %r0_index_11 : tensor<1x16xi32> to tensor<1x16xi64> loc(#loc63)
+      %tmp0_13 = tt.splat %ks0 : i64 -> tensor<1x16xi64> loc(#loc63)
+      %tmp0_14 = arith.muli %tmp0_13, %tmp0 : tensor<1x16xi64> loc(#loc63)
+      %tmp0_15 = tt.broadcast %x0_7 : tensor<64x1xi64> -> tensor<64x16xi64> loc(#loc64)
+      %tmp0_16 = tt.broadcast %tmp0_14 : tensor<1x16xi64> -> tensor<64x16xi64> loc(#loc64)
+      %tmp0_17 = arith.addi %tmp0_15, %tmp0_16 : tensor<64x16xi64> loc(#loc64)
+      %tmp0_18 = arith.muli %ks0, %ks1 : i64 loc(#loc65)
+      %tmp0_19 = tt.splat %tmp0_18 : i64 -> tensor<64x1xi64> loc(#loc66)
+      %tmp0_20 = arith.muli %tmp0_19, %x1 : tensor<64x1xi64> loc(#loc66)
+      %tmp0_21 = tt.broadcast %tmp0_20 : tensor<64x1xi64> -> tensor<64x16xi64> loc(#loc67)
+      %tmp0_22 = arith.addi %tmp0_17, %tmp0_21 : tensor<64x16xi64> loc(#loc67)
+      %tmp0_23 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<64x16x!tt.ptr<i32>> loc(#loc68)
+      %tmp0_24 = tt.addptr %tmp0_23, %tmp0_22 : tensor<64x16x!tt.ptr<i32>>, tensor<64x16xi64> loc(#loc68)
+      %tmp0_25 = tt.broadcast %r0_mask_12 : tensor<1x16xi1> -> tensor<64x16xi1> loc(#loc69)
+      %tmp0_26 = tt.broadcast %xmask_4 : tensor<64x1xi1> -> tensor<64x16xi1> loc(#loc69)
+      %tmp0_27 = arith.andi %tmp0_25, %tmp0_26 : tensor<64x16xi1> loc(#loc69)
+      %tmp0_28 = tt.load %tmp0_24, %tmp0_27, %cst evictionPolicy = evict_last : tensor<64x16x!tt.ptr<i32>> loc(#loc70)
+      %tmp1 = arith.extsi %tmp0_28 : tensor<64x16xi32> to tensor<64x16xi64> loc(#loc71)
+      %tmp4 = arith.addi %_tmp3_10, %tmp1 : tensor<64x16xi64> loc(#loc72)
+      %_tmp3_29 = arith.select %tmp0_27, %tmp4, %_tmp3_10 : tensor<64x16xi1>, tensor<64x16xi64> loc(#loc73)
+      scf.yield %_tmp3_29 : tensor<64x16xi64> loc(#loc27)
+    } loc(#loc60)
+    %tmp3 = "tt.reduce"(%_tmp3_8) <{axis = 1 : i32}> ({
+    ^bb0(%tmp3_10: i64 loc(callsite(#loc1 at #loc74)), %tmp3_11: i64 loc(callsite(#loc1 at #loc74))):
+      %tmp3_12 = arith.addi %tmp3_10, %tmp3_11 : i64 loc(#loc80)
+      tt.reduce.return %tmp3_12 : i64 loc(#loc78)
+    }) : (tensor<64x16xi64>) -> tensor<64xi64> loc(#loc78)
+    %tmp3_9 = tt.expand_dims %tmp3 {axis = 1 : i32} : tensor<64xi64> -> tensor<64x1xi64> loc(#loc75)
+    %tmp5 = arith.trunci %tmp3_9 : tensor<64x1xi64> to tensor<64x1xi32> loc(#loc76)
+    %0 = arith.cmpi sle, %ks0, %c1_i64 : i64 loc(#loc33)
+    %1 = arith.cmpi sgt, %ks0, %c1_i64 : i64 loc(#loc34)
+    %2 = arith.extui %1 : i1 to i64 loc(#loc35)
+    %3 = arith.muli %ks0, %2 : i64 loc(#loc35)
+    %4 = arith.extui %0 : i1 to i64 loc(#loc77)
+    %5 = arith.addi %4, %3 : i64 loc(#loc36)
+    %6 = tt.splat %5 : i64 -> tensor<64x1xi64> loc(#loc38)
+    %7 = arith.muli %x1, %6 : tensor<64x1xi64> loc(#loc38)
+    %8 = arith.addi %x0_7, %7 : tensor<64x1xi64> loc(#loc39)
+    %9 = tt.splat %out_ptr1 : !tt.ptr<i32> -> tensor<64x1x!tt.ptr<i32>> loc(#loc40)
+    %10 = tt.addptr %9, %8 : tensor<64x1x!tt.ptr<i32>>, tensor<64x1xi64> loc(#loc40)
+    tt.store %10, %tmp5, %xmask_4 : tensor<64x1x!tt.ptr<i32>> loc(#loc41)
+    tt.return loc(#loc42)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":30:40)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":28:43)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:28)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":21:33)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:36)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:44)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":22:23)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":23:21)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":24:27)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":24:37)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":26:19)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":27:19)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":31:31)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":32:29)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:43)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:39)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:54)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:58)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:50)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:34)
+#loc22 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:73)
+#loc23 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":36:63)
+#loc24 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":37:23)
+#loc25 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":39:23)
+#loc26 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:48)
+#loc27 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":40:8)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc31 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":41:28)
+#loc32 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":42:19)
+#loc33 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:49)
+#loc34 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:75)
+#loc35 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:66)
+#loc36 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:57)
+#loc37 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:41)
+#loc38 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:34)
+#loc39 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:30)
+#loc40 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:25)
+#loc41 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:88)
+#loc42 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/wh/cwhgu2bzfksecbm4int3okkh3ph54kdms6zfad7ngpfzq6h2owww.py":43:4)
+#loc49 = loc("_tmp3"(#loc3))
+#loc50 = loc("xoffset"(#loc4))
+#loc51 = loc("xoffset"(#loc5))
+#loc52 = loc("xindex"(#loc6))
+#loc53 = loc("xindex"(#loc7))
+#loc54 = loc("xindex"(#loc8))
+#loc55 = loc("xmask"(#loc9))
+#loc56 = loc("r0_base"(#loc10))
+#loc57 = loc("r0_base"(#loc11))
+#loc58 = loc("x0"(#loc12))
+#loc59 = loc("x1"(#loc13))
+#loc60 = loc("_tmp3"(#loc2))
+#loc61 = loc("r0_index"(#loc14))
+#loc62 = loc("r0_mask"(#loc15))
+#loc63 = loc("tmp0"(#loc16))
+#loc64 = loc("tmp0"(#loc17))
+#loc65 = loc("tmp0"(#loc18))
+#loc66 = loc("tmp0"(#loc19))
+#loc67 = loc("tmp0"(#loc20))
+#loc68 = loc("tmp0"(#loc21))
+#loc69 = loc("tmp0"(#loc22))
+#loc70 = loc("tmp0"(#loc23))
+#loc71 = loc("tmp1"(#loc24))
+#loc72 = loc("tmp4"(#loc25))
+#loc73 = loc("_tmp3"(#loc26))
+#loc75 = loc("tmp3"(#loc31))
+#loc76 = loc("tmp5"(#loc32))
+#loc77 = loc(fused[#loc36, #loc37])
+#loc78 = loc(callsite(#loc28 at #loc74))
+#loc80 = loc(callsite(#loc30 at #loc78))

SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/__grp__triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.source", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttir", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttgir", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.llir", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ptx", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.cubin", "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.json

@@ -0,0 +1 @@
+{"hash": "e50c6a987142dbb9ad0869ff13cb48e2cad8442b31ebedd676f284f006fc5f0a", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 4, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 2048, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3"}

SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttgir

@@ -0,0 +1,841 @@
+#blocked = #ttg.blocked<{sizePerThread = [1, 1], threadsPerWarp = [32, 1], warpsPerCTA = [1, 4], order = [0, 1]}>
+#blocked1 = #ttg.blocked<{sizePerThread = [1, 4], threadsPerWarp = [8, 4], warpsPerCTA = [4, 1], order = [1, 0]}>
+#linear = #ttg.linear<{register = [[0, 4], [0, 8]], lane = [[1, 0], [2, 0], [4, 0], [8, 0], [16, 0]], warp = [[0, 1], [0, 2]], block = []}>
+#linear1 = #ttg.linear<{register = [[2, 0, 0], [4, 0, 0]], lane = [[8, 0, 0], [16, 0, 0], [32, 0, 0], [64, 0, 0], [128, 0, 0]], warp = [[0, 1, 0], [1, 0, 0]], block = []}>
+#linear2 = #ttg.linear<{register = [[1, 0, 0], [2, 0, 0]], lane = [[4, 0, 0], [8, 0, 0], [16, 0, 0], [32, 0, 0], [64, 0, 0]], warp = [[0, 0, 1], [0, 1, 0]], block = []}>
+#linear3 = #ttg.linear<{register = [[0, 1, 0], [1, 0, 0]], lane = [[2, 0, 0], [4, 0, 0], [8, 0, 0], [16, 0, 0], [32, 0, 0]], warp = [[0, 0, 1], [0, 0, 2]], block = []}>
+#linear4 = #ttg.linear<{register = [[0, 0, 4], [0, 1, 0]], lane = [[1, 0, 0], [2, 0, 0], [4, 0, 0], [8, 0, 0], [16, 0, 0]], warp = [[0, 0, 1], [0, 0, 2]], block = []}>
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":18:0)
+#loc1 = loc(unknown)
+#loc19 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":662:12)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":41:67)
+#loc24 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":634:73)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:51)
+#loc33 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:53)
+#loc42 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:50)
+#loc47 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:51)
+#loc67 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":45:26)
+#loc77 = loc("in_ptr0"(#loc))
+#loc78 = loc("out_ptr2"(#loc))
+#loc79 = loc("out_ptr3"(#loc))
+#loc80 = loc("xnumel"(#loc))
+#loc81 = loc("r0_numel"(#loc))
+#loc99 = loc(callsite(#loc19 at #loc20))
+#loc105 = loc("ileft"(#loc28))
+#loc109 = loc("iright"(#loc33))
+#loc118 = loc("left_idx"(#loc42))
+#loc123 = loc("right_idx"(#loc47))
+#loc143 = loc("tmp11"(#loc67))
+#loc149 = loc(callsite(#loc24 at #loc99))
+#loc153 = loc(callsite(#loc1 at #loc143))
+#loc157 = loc(callsite(#loc105 at #loc149))
+#loc161 = loc(callsite(#loc109 at #loc149))
+#loc169 = loc(callsite(#loc118 at #loc149))
+#loc174 = loc(callsite(#loc123 at #loc149))
+#loc194 = loc(callsite(#loc1 at #loc157))
+#loc196 = loc(callsite(#loc1 at #loc161))
+#loc199 = loc(callsite(#loc1 at #loc169))
+#loc202 = loc(callsite(#loc1 at #loc174))
+module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 4 : i32, ttg.target = "cuda:90", "ttg.threads-per-warp" = 32 : i32} {
+  tt.func public @triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr2: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr2"(#loc)), %out_ptr3: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr3"(#loc)), %xnumel: i32 {tt.divisibility = 16 : i32} loc("xnumel"(#loc)), %r0_numel: i32 {tt.divisibility = 16 : i32} loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<0> : tensor<32x16xi32, #linear> loc(#loc1)
+    %cst_0 = arith.constant dense<0> : tensor<32x16xi64, #blocked> loc(#loc1)
+    %c32_i32 = arith.constant 32 : i32 loc(#loc1)
+    %cst_1 = arith.constant dense<128> : tensor<32x1xi32, #blocked> loc(#loc1)
+    %cst_2 = arith.constant dense<128> : tensor<32x1xi32, #blocked1> loc(#loc1)
+    %cst_3 = arith.constant dense<16> : tensor<32x1xi32, #blocked> loc(#loc1)
+    %cst_4 = arith.constant dense<16> : tensor<32x1xi32, #blocked1> loc(#loc1)
+    %cst_5 = arith.constant dense<17> : tensor<1x16xi32, #blocked> loc(#loc1)
+    %cst_6 = arith.constant dense<272> : tensor<32x1xi32, #blocked> loc(#loc1)
+    %cst_7 = arith.constant dense<1> : tensor<1x2x1xi32, #linear1> loc(#loc1)
+    %cst_8 = arith.constant dense<1> : tensor<1x2x1xi32, #linear2> loc(#loc1)
+    %cst_9 = arith.constant dense<1> : tensor<1x2x1xi32, #linear3> loc(#loc1)
+    %cst_10 = arith.constant dense<1> : tensor<1x2x1xi32, #linear4> loc(#loc1)
+    %cst_11 = arith.constant dense<0> : tensor<32x16xi32, #blocked> loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc82)
+    %xoffset_12 = arith.muli %xoffset, %c32_i32 : i32 loc(#loc83)
+    %xindex = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc84)
+    %xindex_13 = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked1}>> loc(#loc84)
+    %xindex_14 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<32x1xi32, #blocked> loc(#loc84)
+    %xindex_15 = tt.expand_dims %xindex_13 {axis = 1 : i32} : tensor<32xi32, #ttg.slice<{dim = 1, parent = #blocked1}>> -> tensor<32x1xi32, #blocked1> loc(#loc84)
+    %xindex_16 = tt.splat %xoffset_12 : i32 -> tensor<32x1xi32, #blocked> loc(#loc85)
+    %xindex_17 = tt.splat %xoffset_12 : i32 -> tensor<32x1xi32, #blocked1> loc(#loc85)
+    %xindex_18 = arith.addi %xindex_16, %xindex_14 : tensor<32x1xi32, #blocked> loc(#loc85)
+    %xindex_19 = arith.addi %xindex_17, %xindex_15 : tensor<32x1xi32, #blocked1> loc(#loc85)
+    %xmask = arith.cmpi slt, %xindex_18, %cst_1 : tensor<32x1xi32, #blocked> loc(#loc86)
+    %xmask_20 = arith.cmpi slt, %xindex_19, %cst_2 : tensor<32x1xi32, #blocked1> loc(#loc86)
+    %r0_index = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> loc(#loc87)
+    %r0_index_21 = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #linear}>> loc(#loc87)
+    %r0_index_22 = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked1}>> loc(#loc87)
+    %r0_index_23 = tt.expand_dims %r0_index {axis = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked}>> -> tensor<1x16xi32, #blocked> loc(#loc87)
+    %r0_index_24 = tt.expand_dims %r0_index_21 {axis = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #linear}>> -> tensor<1x16xi32, #linear> loc(#loc87)
+    %r0_index_25 = tt.expand_dims %r0_index_22 {axis = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked1}>> -> tensor<1x16xi32, #blocked1> loc(#loc87)
+    %x0 = arith.remsi %xindex_18, %cst_3 : tensor<32x1xi32, #blocked> loc(#loc88)
+    %x1 = arith.divsi %xindex_18, %cst_3 : tensor<32x1xi32, #blocked> loc(#loc89)
+    %tmp0 = arith.muli %r0_index_23, %cst_5 : tensor<1x16xi32, #blocked> loc(#loc90)
+    %tmp0_26 = tt.broadcast %x0 : tensor<32x1xi32, #blocked> -> tensor<32x16xi32, #blocked> loc(#loc91)
+    %tmp0_27 = tt.broadcast %tmp0 : tensor<1x16xi32, #blocked> -> tensor<32x16xi32, #blocked> loc(#loc91)
+    %tmp0_28 = arith.addi %tmp0_26, %tmp0_27 : tensor<32x16xi32, #blocked> loc(#loc91)
+    %tmp0_29 = arith.muli %x1, %cst_6 : tensor<32x1xi32, #blocked> loc(#loc92)
+    %tmp0_30 = tt.broadcast %tmp0_29 : tensor<32x1xi32, #blocked> -> tensor<32x16xi32, #blocked> loc(#loc93)
+    %tmp0_31 = arith.addi %tmp0_28, %tmp0_30 : tensor<32x16xi32, #blocked> loc(#loc93)
+    %tmp0_32 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>, #blocked> loc(#loc94)
+    %tmp0_33 = tt.addptr %tmp0_32, %tmp0_31 : tensor<32x16x!tt.ptr<i32>, #blocked>, tensor<32x16xi32, #blocked> loc(#loc94)
+    %tmp0_34 = tt.broadcast %xmask : tensor<32x1xi1, #blocked> -> tensor<32x16xi1, #blocked> loc(#loc95)
+    %tmp0_35 = tt.broadcast %xmask_20 : tensor<32x1xi1, #blocked1> -> tensor<32x16xi1, #blocked1> loc(#loc95)
+    %tmp0_36 = tt.load %tmp0_33, %tmp0_34, %cst_11 : tensor<32x16x!tt.ptr<i32>, #blocked> loc(#loc95)
+    %tmp2 = arith.trunci %r0_index_24 : tensor<1x16xi32, #linear> to tensor<1x16xi16, #linear> loc(#loc96)
+    %tmp4 = tt.broadcast %tmp2 : tensor<1x16xi16, #linear> -> tensor<32x16xi16, #linear> loc(#loc97)
+    %flip = tt.make_range {end = 2 : i32, start = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear2}>}>> loc(#loc146)
+    %flip_37 = tt.make_range {end = 2 : i32, start = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear1}>}>> loc(#loc146)
+    %flip_38 = tt.make_range {end = 2 : i32, start = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear3}>}>> loc(#loc146)
+    %flip_39 = tt.make_range {end = 2 : i32, start = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear4}>}>> loc(#loc146)
+    %flip_40 = tt.expand_dims %flip {axis = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear2}>}>> -> tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear2}>> loc(#loc146)
+    %flip_41 = tt.expand_dims %flip_37 {axis = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear1}>}>> -> tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear1}>> loc(#loc146)
+    %flip_42 = tt.expand_dims %flip_38 {axis = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear3}>}>> -> tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear3}>> loc(#loc146)
+    %flip_43 = tt.expand_dims %flip_39 {axis = 0 : i32} : tensor<2xi32, #ttg.slice<{dim = 0, parent = #ttg.slice<{dim = 2, parent = #linear4}>}>> -> tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear4}>> loc(#loc146)
+    %flip_44 = tt.expand_dims %flip_40 {axis = 2 : i32} : tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear2}>> -> tensor<1x2x1xi32, #linear2> loc(#loc146)
+    %flip_45 = tt.expand_dims %flip_41 {axis = 2 : i32} : tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear1}>> -> tensor<1x2x1xi32, #linear1> loc(#loc146)
+    %flip_46 = tt.expand_dims %flip_42 {axis = 2 : i32} : tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear3}>> -> tensor<1x2x1xi32, #linear3> loc(#loc146)
+    %flip_47 = tt.expand_dims %flip_43 {axis = 2 : i32} : tensor<1x2xi32, #ttg.slice<{dim = 2, parent = #linear4}>> -> tensor<1x2x1xi32, #linear4> loc(#loc146)
+    %flip_48 = tt.broadcast %flip_44 : tensor<1x2x1xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc147)
+    %flip_49 = tt.reshape %flip_48 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #blocked> loc(#loc148)
+    %flip_50 = tt.reshape %flip_48 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc148)
+    %y = tt.reshape %tmp0_36 : tensor<32x16xi32, #blocked> -> tensor<256x2x1xi32, #linear1> loc(#loc154)
+    %left_mask = arith.subi %cst_7, %flip_45 : tensor<1x2x1xi32, #linear1> loc(#loc155)
+    %left_mask_51 = arith.subi %cst_8, %flip_44 : tensor<1x2x1xi32, #linear2> loc(#loc155)
+    %left_mask_52 = arith.subi %cst_9, %flip_46 : tensor<1x2x1xi32, #linear3> loc(#loc155)
+    %left_mask_53 = arith.subi %cst_10, %flip_47 : tensor<1x2x1xi32, #linear4> loc(#loc155)
+    %ileft = tt.broadcast %left_mask : tensor<1x2x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc156)
+    %ileft_54 = arith.muli %y, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc156)
+    %ileft_55 = "tt.reduce"(%ileft_54) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc193)
+    %ileft_56 = tt.expand_dims %ileft_55 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc158)
+    %ileft_57 = tt.broadcast %ileft_56 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc159)
+    %iright = tt.broadcast %flip_45 : tensor<1x2x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc160)
+    %iright_58 = arith.muli %y, %iright : tensor<256x2x1xi32, #linear1> loc(#loc160)
+    %iright_59 = "tt.reduce"(%iright_58) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc195)
+    %iright_60 = tt.expand_dims %iright_59 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc162)
+    %iright_61 = tt.broadcast %iright_60 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc163)
+    %ileft_62 = tt.reshape %ileft_57 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #blocked> loc(#loc164)
+    %ileft_63 = tt.reshape %ileft_57 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_64 = tt.reshape %iright_61 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #blocked> loc(#loc165)
+    %iright_65 = tt.reshape %iright_61 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx = tt.reshape %tmp4 : tensor<32x16xi16, #linear> -> tensor<256x2x1xi16, #linear1> loc(#loc166)
+    %left_idx = arith.trunci %left_mask : tensor<1x2x1xi32, #linear1> to tensor<1x2x1xi16, #linear1> loc(#loc167)
+    %left_idx_66 = tt.broadcast %left_idx : tensor<1x2x1xi16, #linear1> -> tensor<256x2x1xi16, #linear1> loc(#loc168)
+    %left_idx_67 = arith.muli %y_idx, %left_idx_66 : tensor<256x2x1xi16, #linear1> loc(#loc168)
+    %input = arith.extsi %left_idx_67 : tensor<256x2x1xi16, #linear1> to tensor<256x2x1xi32, #linear1> loc(#loc197)
+    %left_idx_68 = "tt.reduce"(%input) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc198)
+    %left_idx_69 = tt.expand_dims %left_idx_68 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc170)
+    %left_idx_70 = tt.broadcast %left_idx_69 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc171)
+    %right_idx = arith.trunci %flip_45 : tensor<1x2x1xi32, #linear1> to tensor<1x2x1xi16, #linear1> loc(#loc172)
+    %right_idx_71 = tt.broadcast %right_idx : tensor<1x2x1xi16, #linear1> -> tensor<256x2x1xi16, #linear1> loc(#loc173)
+    %right_idx_72 = arith.muli %y_idx, %right_idx_71 : tensor<256x2x1xi16, #linear1> loc(#loc173)
+    %input_73 = arith.extsi %right_idx_72 : tensor<256x2x1xi16, #linear1> to tensor<256x2x1xi32, #linear1> loc(#loc200)
+    %right_idx_74 = "tt.reduce"(%input_73) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc201)
+    %right_idx_75 = tt.expand_dims %right_idx_74 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc175)
+    %right_idx_76 = tt.broadcast %right_idx_75 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc176)
+    %left_idx_77 = tt.reshape %left_idx_70 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #blocked> loc(#loc177)
+    %left_idx_78 = tt.reshape %left_idx_70 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_79 = tt.reshape %right_idx_76 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #blocked> loc(#loc178)
+    %right_idx_80 = tt.reshape %right_idx_76 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond = arith.cmpi slt, %ileft_62, %iright_64 : tensor<32x16xi32, #blocked> loc(#loc179)
+    %cond_81 = arith.cmpi slt, %ileft_63, %iright_65 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq = arith.cmpi eq, %ileft_62, %iright_64 : tensor<32x16xi32, #blocked> loc(#loc180)
+    %eq_82 = arith.cmpi eq, %ileft_63, %iright_65 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_83 = arith.cmpi sgt, %left_idx_77, %right_idx_79 : tensor<32x16xi32, #blocked> loc(#loc181)
+    %cond_84 = arith.cmpi sgt, %left_idx_78, %right_idx_80 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_85 = arith.andi %eq, %cond_83 : tensor<32x16xi1, #blocked> loc(#loc182)
+    %cond_86 = arith.andi %eq_82, %cond_84 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_87 = arith.ori %cond, %cond_85 : tensor<32x16xi1, #blocked> loc(#loc183)
+    %cond_88 = arith.ori %cond_81, %cond_86 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_89 = arith.extui %cond_87 : tensor<32x16xi1, #blocked> to tensor<32x16xi32, #blocked> loc(#loc184)
+    %cond_90 = arith.extui %cond_88 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_91 = arith.xori %cond_89, %flip_49 : tensor<32x16xi32, #blocked> loc(#loc184)
+    %cond_92 = arith.xori %cond_90, %flip_50 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_93 = arith.cmpi ne, %cond_91, %cst_11 : tensor<32x16xi32, #blocked> loc(#loc185)
+    %cond_94 = arith.cmpi ne, %cond_92, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret = arith.xori %ileft_62, %iright_64 : tensor<32x16xi32, #blocked> loc(#loc186)
+    %ret_95 = arith.select %cond_93, %ret, %cst_11 : tensor<32x16xi1, #blocked>, tensor<32x16xi32, #blocked> loc(#loc187)
+    %ret_96 = arith.xori %tmp0_36, %ret_95 : tensor<32x16xi32, #blocked> loc(#loc188)
+    %ret_97 = ttg.convert_layout %ret_96 : tensor<32x16xi32, #blocked> -> tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs = arith.xori %left_idx_78, %right_idx_80 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_98 = arith.select %cond_94, %new_idxs, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_99 = arith.extsi %tmp2 : tensor<1x16xi16, #linear> to tensor<1x16xi32, #linear> loc(#loc191)
+    %new_idxs_100 = tt.broadcast %new_idxs_99 : tensor<1x16xi32, #linear> -> tensor<32x16xi32, #linear> loc(#loc191)
+    %new_idxs_101 = arith.xori %new_idxs_100, %new_idxs_98 : tensor<32x16xi32, #linear> loc(#loc191)
+    %flip_102 = tt.broadcast %flip_46 : tensor<1x2x1xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc147)
+    %flip_103 = tt.reshape %flip_102 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc148)
+    %y_104 = tt.reshape %ret_96 : tensor<32x16xi32, #blocked> -> tensor<128x2x2xi32, #linear2> loc(#loc154)
+    %ileft_105 = tt.broadcast %left_mask_51 : tensor<1x2x1xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc156)
+    %ileft_106 = arith.muli %y_104, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc156)
+    %ileft_107 = "tt.reduce"(%ileft_106) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc193)
+    %ileft_108 = tt.expand_dims %ileft_107 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc158)
+    %ileft_109 = tt.broadcast %ileft_108 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc159)
+    %iright_110 = arith.muli %y_104, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc160)
+    %iright_111 = "tt.reduce"(%iright_110) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc195)
+    %iright_112 = tt.expand_dims %iright_111 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc162)
+    %iright_113 = tt.broadcast %iright_112 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc163)
+    %ileft_114 = tt.reshape %ileft_109 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_115 = tt.reshape %iright_113 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_116 = tt.reshape %new_idxs_101 : tensor<32x16xi32, #linear> -> tensor<128x2x2xi32, #linear2> loc(#loc166)
+    %left_idx_117 = arith.muli %y_idx_116, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc168)
+    %left_idx_118 = "tt.reduce"(%left_idx_117) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc198)
+    %left_idx_119 = tt.expand_dims %left_idx_118 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc170)
+    %left_idx_120 = tt.broadcast %left_idx_119 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc171)
+    %right_idx_121 = arith.muli %y_idx_116, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc173)
+    %right_idx_122 = "tt.reduce"(%right_idx_121) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc201)
+    %right_idx_123 = tt.expand_dims %right_idx_122 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc175)
+    %right_idx_124 = tt.broadcast %right_idx_123 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc176)
+    %left_idx_125 = tt.reshape %left_idx_120 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_126 = tt.reshape %right_idx_124 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_127 = arith.cmpi slt, %ileft_114, %iright_115 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_128 = arith.cmpi eq, %ileft_114, %iright_115 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_129 = arith.cmpi sgt, %left_idx_125, %right_idx_126 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_130 = arith.andi %eq_128, %cond_129 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_131 = arith.ori %cond_127, %cond_130 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_132 = arith.extui %cond_131 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_133 = arith.xori %cond_132, %flip_103 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_134 = arith.cmpi ne, %cond_133, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret_135 = arith.xori %ileft_114, %iright_115 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_136 = arith.select %cond_134, %ret_135, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_137 = arith.xori %ret_97, %ret_136 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_138 = arith.xori %left_idx_125, %right_idx_126 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_139 = arith.select %cond_134, %new_idxs_138, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_140 = arith.xori %new_idxs_101, %new_idxs_139 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_141 = tt.reshape %ret_137 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc154)
+    %ileft_142 = arith.muli %y_141, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc156)
+    %ileft_143 = "tt.reduce"(%ileft_142) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc193)
+    %ileft_144 = tt.expand_dims %ileft_143 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc158)
+    %ileft_145 = tt.broadcast %ileft_144 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc159)
+    %iright_146 = arith.muli %y_141, %iright : tensor<256x2x1xi32, #linear1> loc(#loc160)
+    %iright_147 = "tt.reduce"(%iright_146) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc195)
+    %iright_148 = tt.expand_dims %iright_147 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc162)
+    %iright_149 = tt.broadcast %iright_148 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc163)
+    %ileft_150 = tt.reshape %ileft_145 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_151 = tt.reshape %iright_149 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_152 = tt.reshape %new_idxs_140 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc166)
+    %left_idx_153 = arith.muli %y_idx_152, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc168)
+    %left_idx_154 = "tt.reduce"(%left_idx_153) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc198)
+    %left_idx_155 = tt.expand_dims %left_idx_154 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc170)
+    %left_idx_156 = tt.broadcast %left_idx_155 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc171)
+    %right_idx_157 = arith.muli %y_idx_152, %iright : tensor<256x2x1xi32, #linear1> loc(#loc173)
+    %right_idx_158 = "tt.reduce"(%right_idx_157) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc201)
+    %right_idx_159 = tt.expand_dims %right_idx_158 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc175)
+    %right_idx_160 = tt.broadcast %right_idx_159 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc176)
+    %left_idx_161 = tt.reshape %left_idx_156 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_162 = tt.reshape %right_idx_160 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_163 = arith.cmpi slt, %ileft_150, %iright_151 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_164 = arith.cmpi eq, %ileft_150, %iright_151 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_165 = arith.cmpi sgt, %left_idx_161, %right_idx_162 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_166 = arith.andi %eq_164, %cond_165 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_167 = arith.ori %cond_163, %cond_166 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_168 = arith.extui %cond_167 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_169 = arith.xori %cond_168, %flip_103 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_170 = arith.cmpi ne, %cond_169, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret_171 = arith.xori %ileft_150, %iright_151 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_172 = arith.select %cond_170, %ret_171, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_173 = arith.xori %ret_137, %ret_172 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_174 = arith.xori %left_idx_161, %right_idx_162 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_175 = arith.select %cond_170, %new_idxs_174, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_176 = arith.xori %new_idxs_140, %new_idxs_175 : tensor<32x16xi32, #linear> loc(#loc191)
+    %flip_177 = tt.broadcast %flip_47 : tensor<1x2x1xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc147)
+    %flip_178 = tt.reshape %flip_177 : tensor<32x2x8xi32, #linear4> -> tensor<32x16xi32, #linear> loc(#loc148)
+    %y_179 = tt.reshape %ret_173 : tensor<32x16xi32, #linear> -> tensor<64x2x4xi32, #linear3> loc(#loc154)
+    %ileft_180 = tt.broadcast %left_mask_52 : tensor<1x2x1xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc156)
+    %ileft_181 = arith.muli %y_179, %ileft_180 : tensor<64x2x4xi32, #linear3> loc(#loc156)
+    %ileft_182 = "tt.reduce"(%ileft_181) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc193)
+    %ileft_183 = tt.expand_dims %ileft_182 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc158)
+    %ileft_184 = tt.broadcast %ileft_183 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc159)
+    %iright_185 = arith.muli %y_179, %flip_102 : tensor<64x2x4xi32, #linear3> loc(#loc160)
+    %iright_186 = "tt.reduce"(%iright_185) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc195)
+    %iright_187 = tt.expand_dims %iright_186 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc162)
+    %iright_188 = tt.broadcast %iright_187 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc163)
+    %ileft_189 = tt.reshape %ileft_184 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_190 = tt.reshape %iright_188 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_191 = tt.reshape %new_idxs_176 : tensor<32x16xi32, #linear> -> tensor<64x2x4xi32, #linear3> loc(#loc166)
+    %left_idx_192 = arith.muli %y_idx_191, %ileft_180 : tensor<64x2x4xi32, #linear3> loc(#loc168)
+    %left_idx_193 = "tt.reduce"(%left_idx_192) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc198)
+    %left_idx_194 = tt.expand_dims %left_idx_193 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc170)
+    %left_idx_195 = tt.broadcast %left_idx_194 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc171)
+    %right_idx_196 = arith.muli %y_idx_191, %flip_102 : tensor<64x2x4xi32, #linear3> loc(#loc173)
+    %right_idx_197 = "tt.reduce"(%right_idx_196) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc201)
+    %right_idx_198 = tt.expand_dims %right_idx_197 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc175)
+    %right_idx_199 = tt.broadcast %right_idx_198 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc176)
+    %left_idx_200 = tt.reshape %left_idx_195 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_201 = tt.reshape %right_idx_199 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_202 = arith.cmpi slt, %ileft_189, %iright_190 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_203 = arith.cmpi eq, %ileft_189, %iright_190 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_204 = arith.cmpi sgt, %left_idx_200, %right_idx_201 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_205 = arith.andi %eq_203, %cond_204 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_206 = arith.ori %cond_202, %cond_205 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_207 = arith.extui %cond_206 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_208 = arith.xori %cond_207, %flip_178 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_209 = arith.cmpi ne, %cond_208, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret_210 = arith.xori %ileft_189, %iright_190 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_211 = arith.select %cond_209, %ret_210, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_212 = arith.xori %ret_173, %ret_211 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_213 = arith.xori %left_idx_200, %right_idx_201 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_214 = arith.select %cond_209, %new_idxs_213, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_215 = arith.xori %new_idxs_176, %new_idxs_214 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_216 = tt.reshape %ret_212 : tensor<32x16xi32, #linear> -> tensor<128x2x2xi32, #linear2> loc(#loc154)
+    %ileft_217 = arith.muli %y_216, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc156)
+    %ileft_218 = "tt.reduce"(%ileft_217) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc193)
+    %ileft_219 = tt.expand_dims %ileft_218 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc158)
+    %ileft_220 = tt.broadcast %ileft_219 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc159)
+    %iright_221 = arith.muli %y_216, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc160)
+    %iright_222 = "tt.reduce"(%iright_221) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc195)
+    %iright_223 = tt.expand_dims %iright_222 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc162)
+    %iright_224 = tt.broadcast %iright_223 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc163)
+    %ileft_225 = tt.reshape %ileft_220 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_226 = tt.reshape %iright_224 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_227 = tt.reshape %new_idxs_215 : tensor<32x16xi32, #linear> -> tensor<128x2x2xi32, #linear2> loc(#loc166)
+    %left_idx_228 = arith.muli %y_idx_227, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc168)
+    %left_idx_229 = "tt.reduce"(%left_idx_228) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc198)
+    %left_idx_230 = tt.expand_dims %left_idx_229 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc170)
+    %left_idx_231 = tt.broadcast %left_idx_230 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc171)
+    %right_idx_232 = arith.muli %y_idx_227, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc173)
+    %right_idx_233 = "tt.reduce"(%right_idx_232) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc201)
+    %right_idx_234 = tt.expand_dims %right_idx_233 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc175)
+    %right_idx_235 = tt.broadcast %right_idx_234 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc176)
+    %left_idx_236 = tt.reshape %left_idx_231 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_237 = tt.reshape %right_idx_235 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_238 = arith.cmpi slt, %ileft_225, %iright_226 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_239 = arith.cmpi eq, %ileft_225, %iright_226 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_240 = arith.cmpi sgt, %left_idx_236, %right_idx_237 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_241 = arith.andi %eq_239, %cond_240 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_242 = arith.ori %cond_238, %cond_241 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_243 = arith.extui %cond_242 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_244 = arith.xori %cond_243, %flip_178 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_245 = arith.cmpi ne, %cond_244, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret_246 = arith.xori %ileft_225, %iright_226 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_247 = arith.select %cond_245, %ret_246, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_248 = arith.xori %ret_212, %ret_247 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_249 = arith.xori %left_idx_236, %right_idx_237 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_250 = arith.select %cond_245, %new_idxs_249, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_251 = arith.xori %new_idxs_215, %new_idxs_250 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_252 = tt.reshape %ret_248 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc154)
+    %ileft_253 = arith.muli %y_252, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc156)
+    %ileft_254 = "tt.reduce"(%ileft_253) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc193)
+    %ileft_255 = tt.expand_dims %ileft_254 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc158)
+    %ileft_256 = tt.broadcast %ileft_255 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc159)
+    %iright_257 = arith.muli %y_252, %iright : tensor<256x2x1xi32, #linear1> loc(#loc160)
+    %iright_258 = "tt.reduce"(%iright_257) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc195)
+    %iright_259 = tt.expand_dims %iright_258 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc162)
+    %iright_260 = tt.broadcast %iright_259 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc163)
+    %ileft_261 = tt.reshape %ileft_256 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_262 = tt.reshape %iright_260 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_263 = tt.reshape %new_idxs_251 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc166)
+    %left_idx_264 = arith.muli %y_idx_263, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc168)
+    %left_idx_265 = "tt.reduce"(%left_idx_264) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc198)
+    %left_idx_266 = tt.expand_dims %left_idx_265 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc170)
+    %left_idx_267 = tt.broadcast %left_idx_266 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc171)
+    %right_idx_268 = arith.muli %y_idx_263, %iright : tensor<256x2x1xi32, #linear1> loc(#loc173)
+    %right_idx_269 = "tt.reduce"(%right_idx_268) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc201)
+    %right_idx_270 = tt.expand_dims %right_idx_269 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc175)
+    %right_idx_271 = tt.broadcast %right_idx_270 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc176)
+    %left_idx_272 = tt.reshape %left_idx_267 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_273 = tt.reshape %right_idx_271 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_274 = arith.cmpi slt, %ileft_261, %iright_262 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_275 = arith.cmpi eq, %ileft_261, %iright_262 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_276 = arith.cmpi sgt, %left_idx_272, %right_idx_273 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_277 = arith.andi %eq_275, %cond_276 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_278 = arith.ori %cond_274, %cond_277 : tensor<32x16xi1, #linear> loc(#loc183)
+    %cond_279 = arith.extui %cond_278 : tensor<32x16xi1, #linear> to tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_280 = arith.xori %cond_279, %flip_178 : tensor<32x16xi32, #linear> loc(#loc184)
+    %cond_281 = arith.cmpi ne, %cond_280, %cst : tensor<32x16xi32, #linear> loc(#loc185)
+    %ret_282 = arith.xori %ileft_261, %iright_262 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_283 = arith.select %cond_281, %ret_282, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_284 = arith.xori %ret_248, %ret_283 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_285 = arith.xori %left_idx_272, %right_idx_273 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_286 = arith.select %cond_281, %new_idxs_285, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_287 = arith.xori %new_idxs_251, %new_idxs_286 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_288 = tt.reshape %ret_284 : tensor<32x16xi32, #linear> -> tensor<32x2x8xi32, #linear4> loc(#loc154)
+    %ileft_289 = tt.broadcast %left_mask_53 : tensor<1x2x1xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc156)
+    %ileft_290 = arith.muli %y_288, %ileft_289 : tensor<32x2x8xi32, #linear4> loc(#loc156)
+    %ileft_291 = "tt.reduce"(%ileft_290) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<32x2x8xi32, #linear4>) -> tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> loc(#loc193)
+    %ileft_292 = tt.expand_dims %ileft_291 {axis = 1 : i32} : tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> -> tensor<32x1x8xi32, #linear4> loc(#loc158)
+    %ileft_293 = tt.broadcast %ileft_292 : tensor<32x1x8xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc159)
+    %iright_294 = arith.muli %y_288, %flip_177 : tensor<32x2x8xi32, #linear4> loc(#loc160)
+    %iright_295 = "tt.reduce"(%iright_294) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<32x2x8xi32, #linear4>) -> tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> loc(#loc195)
+    %iright_296 = tt.expand_dims %iright_295 {axis = 1 : i32} : tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> -> tensor<32x1x8xi32, #linear4> loc(#loc162)
+    %iright_297 = tt.broadcast %iright_296 : tensor<32x1x8xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc163)
+    %ileft_298 = tt.reshape %ileft_293 : tensor<32x2x8xi32, #linear4> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_299 = tt.reshape %iright_297 : tensor<32x2x8xi32, #linear4> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_300 = tt.reshape %new_idxs_287 : tensor<32x16xi32, #linear> -> tensor<32x2x8xi32, #linear4> loc(#loc166)
+    %left_idx_301 = arith.muli %y_idx_300, %ileft_289 : tensor<32x2x8xi32, #linear4> loc(#loc168)
+    %left_idx_302 = "tt.reduce"(%left_idx_301) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<32x2x8xi32, #linear4>) -> tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> loc(#loc198)
+    %left_idx_303 = tt.expand_dims %left_idx_302 {axis = 1 : i32} : tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> -> tensor<32x1x8xi32, #linear4> loc(#loc170)
+    %left_idx_304 = tt.broadcast %left_idx_303 : tensor<32x1x8xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc171)
+    %right_idx_305 = arith.muli %y_idx_300, %flip_177 : tensor<32x2x8xi32, #linear4> loc(#loc173)
+    %right_idx_306 = "tt.reduce"(%right_idx_305) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<32x2x8xi32, #linear4>) -> tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> loc(#loc201)
+    %right_idx_307 = tt.expand_dims %right_idx_306 {axis = 1 : i32} : tensor<32x8xi32, #ttg.slice<{dim = 1, parent = #linear4}>> -> tensor<32x1x8xi32, #linear4> loc(#loc175)
+    %right_idx_308 = tt.broadcast %right_idx_307 : tensor<32x1x8xi32, #linear4> -> tensor<32x2x8xi32, #linear4> loc(#loc176)
+    %left_idx_309 = tt.reshape %left_idx_304 : tensor<32x2x8xi32, #linear4> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_310 = tt.reshape %right_idx_308 : tensor<32x2x8xi32, #linear4> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_311 = arith.cmpi slt, %ileft_298, %iright_299 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_312 = arith.cmpi eq, %ileft_298, %iright_299 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_313 = arith.cmpi sgt, %left_idx_309, %right_idx_310 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_314 = arith.andi %eq_312, %cond_313 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_315 = arith.ori %cond_311, %cond_314 : tensor<32x16xi1, #linear> loc(#loc183)
+    %ret_316 = arith.xori %ileft_298, %iright_299 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_317 = arith.select %cond_315, %ret_316, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_318 = arith.xori %ret_284, %ret_317 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_319 = arith.xori %left_idx_309, %right_idx_310 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_320 = arith.select %cond_315, %new_idxs_319, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_321 = arith.xori %new_idxs_287, %new_idxs_320 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_322 = tt.reshape %ret_318 : tensor<32x16xi32, #linear> -> tensor<64x2x4xi32, #linear3> loc(#loc154)
+    %ileft_323 = arith.muli %y_322, %ileft_180 : tensor<64x2x4xi32, #linear3> loc(#loc156)
+    %ileft_324 = "tt.reduce"(%ileft_323) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc193)
+    %ileft_325 = tt.expand_dims %ileft_324 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc158)
+    %ileft_326 = tt.broadcast %ileft_325 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc159)
+    %iright_327 = arith.muli %y_322, %flip_102 : tensor<64x2x4xi32, #linear3> loc(#loc160)
+    %iright_328 = "tt.reduce"(%iright_327) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc195)
+    %iright_329 = tt.expand_dims %iright_328 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc162)
+    %iright_330 = tt.broadcast %iright_329 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc163)
+    %ileft_331 = tt.reshape %ileft_326 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_332 = tt.reshape %iright_330 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_333 = tt.reshape %new_idxs_321 : tensor<32x16xi32, #linear> -> tensor<64x2x4xi32, #linear3> loc(#loc166)
+    %left_idx_334 = arith.muli %y_idx_333, %ileft_180 : tensor<64x2x4xi32, #linear3> loc(#loc168)
+    %left_idx_335 = "tt.reduce"(%left_idx_334) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc198)
+    %left_idx_336 = tt.expand_dims %left_idx_335 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc170)
+    %left_idx_337 = tt.broadcast %left_idx_336 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc171)
+    %right_idx_338 = arith.muli %y_idx_333, %flip_102 : tensor<64x2x4xi32, #linear3> loc(#loc173)
+    %right_idx_339 = "tt.reduce"(%right_idx_338) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<64x2x4xi32, #linear3>) -> tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> loc(#loc201)
+    %right_idx_340 = tt.expand_dims %right_idx_339 {axis = 1 : i32} : tensor<64x4xi32, #ttg.slice<{dim = 1, parent = #linear3}>> -> tensor<64x1x4xi32, #linear3> loc(#loc175)
+    %right_idx_341 = tt.broadcast %right_idx_340 : tensor<64x1x4xi32, #linear3> -> tensor<64x2x4xi32, #linear3> loc(#loc176)
+    %left_idx_342 = tt.reshape %left_idx_337 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_343 = tt.reshape %right_idx_341 : tensor<64x2x4xi32, #linear3> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_344 = arith.cmpi slt, %ileft_331, %iright_332 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_345 = arith.cmpi eq, %ileft_331, %iright_332 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_346 = arith.cmpi sgt, %left_idx_342, %right_idx_343 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_347 = arith.andi %eq_345, %cond_346 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_348 = arith.ori %cond_344, %cond_347 : tensor<32x16xi1, #linear> loc(#loc183)
+    %ret_349 = arith.xori %ileft_331, %iright_332 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_350 = arith.select %cond_348, %ret_349, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_351 = arith.xori %ret_318, %ret_350 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_352 = arith.xori %left_idx_342, %right_idx_343 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_353 = arith.select %cond_348, %new_idxs_352, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_354 = arith.xori %new_idxs_321, %new_idxs_353 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_355 = tt.reshape %ret_351 : tensor<32x16xi32, #linear> -> tensor<128x2x2xi32, #linear2> loc(#loc154)
+    %ileft_356 = arith.muli %y_355, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc156)
+    %ileft_357 = "tt.reduce"(%ileft_356) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc193)
+    %ileft_358 = tt.expand_dims %ileft_357 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc158)
+    %ileft_359 = tt.broadcast %ileft_358 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc159)
+    %iright_360 = arith.muli %y_355, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc160)
+    %iright_361 = "tt.reduce"(%iright_360) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc195)
+    %iright_362 = tt.expand_dims %iright_361 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc162)
+    %iright_363 = tt.broadcast %iright_362 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc163)
+    %ileft_364 = tt.reshape %ileft_359 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_365 = tt.reshape %iright_363 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_366 = tt.reshape %new_idxs_354 : tensor<32x16xi32, #linear> -> tensor<128x2x2xi32, #linear2> loc(#loc166)
+    %left_idx_367 = arith.muli %y_idx_366, %ileft_105 : tensor<128x2x2xi32, #linear2> loc(#loc168)
+    %left_idx_368 = "tt.reduce"(%left_idx_367) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc198)
+    %left_idx_369 = tt.expand_dims %left_idx_368 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc170)
+    %left_idx_370 = tt.broadcast %left_idx_369 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc171)
+    %right_idx_371 = arith.muli %y_idx_366, %flip_48 : tensor<128x2x2xi32, #linear2> loc(#loc173)
+    %right_idx_372 = "tt.reduce"(%right_idx_371) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32, #linear2>) -> tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> loc(#loc201)
+    %right_idx_373 = tt.expand_dims %right_idx_372 {axis = 1 : i32} : tensor<128x2xi32, #ttg.slice<{dim = 1, parent = #linear2}>> -> tensor<128x1x2xi32, #linear2> loc(#loc175)
+    %right_idx_374 = tt.broadcast %right_idx_373 : tensor<128x1x2xi32, #linear2> -> tensor<128x2x2xi32, #linear2> loc(#loc176)
+    %left_idx_375 = tt.reshape %left_idx_370 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_376 = tt.reshape %right_idx_374 : tensor<128x2x2xi32, #linear2> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_377 = arith.cmpi slt, %ileft_364, %iright_365 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_378 = arith.cmpi eq, %ileft_364, %iright_365 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_379 = arith.cmpi sgt, %left_idx_375, %right_idx_376 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_380 = arith.andi %eq_378, %cond_379 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_381 = arith.ori %cond_377, %cond_380 : tensor<32x16xi1, #linear> loc(#loc183)
+    %ret_382 = arith.xori %ileft_364, %iright_365 : tensor<32x16xi32, #linear> loc(#loc186)
+    %ret_383 = arith.select %cond_381, %ret_382, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc187)
+    %ret_384 = arith.xori %ret_351, %ret_383 : tensor<32x16xi32, #linear> loc(#loc188)
+    %new_idxs_385 = arith.xori %left_idx_375, %right_idx_376 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_386 = arith.select %cond_381, %new_idxs_385, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_387 = arith.xori %new_idxs_354, %new_idxs_386 : tensor<32x16xi32, #linear> loc(#loc191)
+    %y_388 = tt.reshape %ret_384 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc154)
+    %ileft_389 = arith.muli %y_388, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc156)
+    %ileft_390 = "tt.reduce"(%ileft_389) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_419: i32 loc(callsite(#loc1 at #loc157)), %ileft_420: i32 loc(callsite(#loc1 at #loc157))):
+      %ileft_421 = arith.addi %ileft_419, %ileft_420 : i32 loc(#loc203)
+      tt.reduce.return %ileft_421 : i32 loc(#loc193)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc193)
+    %ileft_391 = tt.expand_dims %ileft_390 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc158)
+    %ileft_392 = tt.broadcast %ileft_391 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc159)
+    %iright_393 = arith.muli %y_388, %iright : tensor<256x2x1xi32, #linear1> loc(#loc160)
+    %iright_394 = "tt.reduce"(%iright_393) <{axis = 1 : i32}> ({
+    ^bb0(%iright_419: i32 loc(callsite(#loc1 at #loc161)), %iright_420: i32 loc(callsite(#loc1 at #loc161))):
+      %iright_421 = arith.addi %iright_419, %iright_420 : i32 loc(#loc204)
+      tt.reduce.return %iright_421 : i32 loc(#loc195)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc195)
+    %iright_395 = tt.expand_dims %iright_394 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc162)
+    %iright_396 = tt.broadcast %iright_395 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc163)
+    %ileft_397 = tt.reshape %ileft_392 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc164)
+    %iright_398 = tt.reshape %iright_396 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc165)
+    %y_idx_399 = tt.reshape %new_idxs_387 : tensor<32x16xi32, #linear> -> tensor<256x2x1xi32, #linear1> loc(#loc166)
+    %left_idx_400 = arith.muli %y_idx_399, %ileft : tensor<256x2x1xi32, #linear1> loc(#loc168)
+    %left_idx_401 = "tt.reduce"(%left_idx_400) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_419: i32 loc(callsite(#loc1 at #loc169)), %left_idx_420: i32 loc(callsite(#loc1 at #loc169))):
+      %left_idx_421 = arith.addi %left_idx_419, %left_idx_420 : i32 loc(#loc205)
+      tt.reduce.return %left_idx_421 : i32 loc(#loc198)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc198)
+    %left_idx_402 = tt.expand_dims %left_idx_401 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc170)
+    %left_idx_403 = tt.broadcast %left_idx_402 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc171)
+    %right_idx_404 = arith.muli %y_idx_399, %iright : tensor<256x2x1xi32, #linear1> loc(#loc173)
+    %right_idx_405 = "tt.reduce"(%right_idx_404) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_419: i32 loc(callsite(#loc1 at #loc174)), %right_idx_420: i32 loc(callsite(#loc1 at #loc174))):
+      %right_idx_421 = arith.addi %right_idx_419, %right_idx_420 : i32 loc(#loc206)
+      tt.reduce.return %right_idx_421 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32, #linear1>) -> tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> loc(#loc201)
+    %right_idx_406 = tt.expand_dims %right_idx_405 {axis = 1 : i32} : tensor<256x1xi32, #ttg.slice<{dim = 1, parent = #linear1}>> -> tensor<256x1x1xi32, #linear1> loc(#loc175)
+    %right_idx_407 = tt.broadcast %right_idx_406 : tensor<256x1x1xi32, #linear1> -> tensor<256x2x1xi32, #linear1> loc(#loc176)
+    %left_idx_408 = tt.reshape %left_idx_403 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc177)
+    %right_idx_409 = tt.reshape %right_idx_407 : tensor<256x2x1xi32, #linear1> -> tensor<32x16xi32, #linear> loc(#loc178)
+    %cond_410 = arith.cmpi slt, %ileft_397, %iright_398 : tensor<32x16xi32, #linear> loc(#loc179)
+    %eq_411 = arith.cmpi eq, %ileft_397, %iright_398 : tensor<32x16xi32, #linear> loc(#loc180)
+    %cond_412 = arith.cmpi sgt, %left_idx_408, %right_idx_409 : tensor<32x16xi32, #linear> loc(#loc181)
+    %cond_413 = arith.andi %eq_411, %cond_412 : tensor<32x16xi1, #linear> loc(#loc182)
+    %cond_414 = arith.ori %cond_410, %cond_413 : tensor<32x16xi1, #linear> loc(#loc183)
+    %new_idxs_415 = arith.xori %left_idx_408, %right_idx_409 : tensor<32x16xi32, #linear> loc(#loc189)
+    %new_idxs_416 = arith.select %cond_414, %new_idxs_415, %cst : tensor<32x16xi1, #linear>, tensor<32x16xi32, #linear> loc(#loc190)
+    %new_idxs_417 = arith.xori %new_idxs_387, %new_idxs_416 : tensor<32x16xi32, #linear> loc(#loc191)
+    %tmp7 = arith.extsi %tmp0_36 : tensor<32x16xi32, #blocked> to tensor<32x16xi64, #blocked> loc(#loc141)
+    %tmp10 = arith.select %tmp0_34, %tmp7, %cst_0 : tensor<32x16xi1, #blocked>, tensor<32x16xi64, #blocked> loc(#loc142)
+    %tmp11 = "tt.reduce"(%tmp10) <{axis = 1 : i32}> ({
+    ^bb0(%tmp11_419: i64 loc(callsite(#loc1 at #loc143)), %tmp11_420: i64 loc(callsite(#loc1 at #loc143))):
+      %tmp11_421 = arith.addi %tmp11_419, %tmp11_420 : i64 loc(#loc192)
+      tt.reduce.return %tmp11_421 : i64 loc(#loc152)
+    }) : (tensor<32x16xi64, #blocked>) -> tensor<32xi64, #ttg.slice<{dim = 1, parent = #blocked}>> loc(#loc152)
+    %tmp11_418 = tt.expand_dims %tmp11 {axis = 1 : i32} : tensor<32xi64, #ttg.slice<{dim = 1, parent = #blocked}>> -> tensor<32x1xi64, #blocked> loc(#loc144)
+    %tmp14 = arith.trunci %tmp11_418 : tensor<32x1xi64, #blocked> to tensor<32x1xi32, #blocked> loc(#loc145)
+    %0 = arith.muli %xindex_19, %cst_4 : tensor<32x1xi32, #blocked1> loc(#loc70)
+    %1 = tt.broadcast %r0_index_25 : tensor<1x16xi32, #blocked1> -> tensor<32x16xi32, #blocked1> loc(#loc71)
+    %2 = tt.broadcast %0 : tensor<32x1xi32, #blocked1> -> tensor<32x16xi32, #blocked1> loc(#loc71)
+    %3 = arith.addi %1, %2 : tensor<32x16xi32, #blocked1> loc(#loc71)
+    %4 = tt.splat %out_ptr2 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>, #blocked1> loc(#loc72)
+    %5 = tt.addptr %4, %3 : tensor<32x16x!tt.ptr<i32>, #blocked1>, tensor<32x16xi32, #blocked1> loc(#loc72)
+    %6 = ttg.convert_layout %new_idxs_417 : tensor<32x16xi32, #linear> -> tensor<32x16xi32, #blocked1> loc(#loc73)
+    tt.store %5, %6, %tmp0_35 : tensor<32x16x!tt.ptr<i32>, #blocked1> loc(#loc73)
+    %7 = tt.splat %out_ptr3 : !tt.ptr<i32> -> tensor<32x1x!tt.ptr<i32>, #blocked> loc(#loc74)
+    %8 = tt.addptr %7, %xindex_18 : tensor<32x1x!tt.ptr<i32>, #blocked>, tensor<32x1xi32, #blocked> loc(#loc74)
+    tt.store %8, %tmp14, %xmask : tensor<32x1x!tt.ptr<i32>, #blocked> loc(#loc75)
+    tt.return loc(#loc76)
+  } loc(#loc)
+} loc(#loc)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":24:28)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":24:33)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":25:44)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":25:23)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":26:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":27:38)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":33:19)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":34:19)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:38)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:35)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:49)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:45)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:30)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:54)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":38:19)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":40:33)
+#loc18 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:44)
+#loc21 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:60)
+#loc22 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:68)
+#loc23 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":533:22)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":537:21)
+#loc26 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:40)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc29 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:65)
+#loc31 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:78)
+#loc32 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:41)
+#loc34 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:67)
+#loc35 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:80)
+#loc36 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":540:30)
+#loc37 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":541:32)
+#loc38 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":546:29)
+#loc39 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:36)
+#loc40 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:23)
+#loc41 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":290:25)
+#loc43 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:53)
+#loc44 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:66)
+#loc45 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:37)
+#loc46 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:23)
+#loc48 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:54)
+#loc49 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:67)
+#loc50 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":553:36)
+#loc51 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":554:38)
+#loc52 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":574:22)
+#loc53 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":591:21)
+#loc54 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:40)
+#loc55 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:29)
+#loc56 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:23)
+#loc57 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":599:19)
+#loc58 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":599:28)
+#loc59 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:38)
+#loc60 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:46)
+#loc61 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:15)
+#loc62 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:48)
+#loc63 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:59)
+#loc64 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:22)
+#loc65 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":42:19)
+#loc66 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":44:34)
+#loc68 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":45:29)
+#loc69 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":48:21)
+#loc70 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:35)
+#loc71 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:32)
+#loc72 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:25)
+#loc73 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:47)
+#loc74 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:25)
+#loc75 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:37)
+#loc76 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:4)
+#loc82 = loc("xoffset"(#loc2))
+#loc83 = loc("xoffset"(#loc3))
+#loc84 = loc("xindex"(#loc4))
+#loc85 = loc("xindex"(#loc5))
+#loc86 = loc("xmask"(#loc6))
+#loc87 = loc("r0_index"(#loc7))
+#loc88 = loc("x0"(#loc8))
+#loc89 = loc("x1"(#loc9))
+#loc90 = loc("tmp0"(#loc10))
+#loc91 = loc("tmp0"(#loc11))
+#loc92 = loc("tmp0"(#loc12))
+#loc93 = loc("tmp0"(#loc13))
+#loc94 = loc("tmp0"(#loc14))
+#loc95 = loc("tmp0"(#loc15))
+#loc96 = loc("tmp2"(#loc16))
+#loc97 = loc("tmp4"(#loc17))
+#loc98 = loc("flip"(#loc18))
+#loc100 = loc("flip"(#loc21))
+#loc101 = loc("flip"(#loc22))
+#loc102 = loc("y"(#loc23))
+#loc103 = loc("left_mask"(#loc25))
+#loc104 = loc("ileft"(#loc26))
+#loc106 = loc("ileft"(#loc30))
+#loc107 = loc("ileft"(#loc31))
+#loc108 = loc("iright"(#loc32))
+#loc110 = loc("iright"(#loc34))
+#loc111 = loc("iright"(#loc35))
+#loc112 = loc("ileft"(#loc36))
+#loc113 = loc("iright"(#loc37))
+#loc114 = loc("y_idx"(#loc38))
+#loc115 = loc("left_idx"(#loc39))
+#loc116 = loc("left_idx"(#loc40))
+#loc117 = loc("input"(#loc41))
+#loc119 = loc("left_idx"(#loc43))
+#loc120 = loc("left_idx"(#loc44))
+#loc121 = loc("right_idx"(#loc45))
+#loc122 = loc("right_idx"(#loc46))
+#loc124 = loc("right_idx"(#loc48))
+#loc125 = loc("right_idx"(#loc49))
+#loc126 = loc("left_idx"(#loc50))
+#loc127 = loc("right_idx"(#loc51))
+#loc128 = loc("cond"(#loc52))
+#loc129 = loc("eq"(#loc53))
+#loc130 = loc("cond"(#loc54))
+#loc131 = loc("cond"(#loc55))
+#loc132 = loc("cond"(#loc56))
+#loc133 = loc("cond"(#loc57))
+#loc134 = loc("cond"(#loc58))
+#loc135 = loc("ret"(#loc59))
+#loc136 = loc("ret"(#loc60))
+#loc137 = loc("ret"(#loc61))
+#loc138 = loc("new_idxs"(#loc62))
+#loc139 = loc("new_idxs"(#loc63))
+#loc140 = loc("new_idxs"(#loc64))
+#loc141 = loc("tmp7"(#loc65))
+#loc142 = loc("tmp10"(#loc66))
+#loc144 = loc("tmp11"(#loc68))
+#loc145 = loc("tmp14"(#loc69))
+#loc146 = loc(callsite(#loc98 at #loc99))
+#loc147 = loc(callsite(#loc100 at #loc99))
+#loc148 = loc(callsite(#loc101 at #loc99))
+#loc150 = loc("cond"(#loc128))
+#loc151 = loc("eq"(#loc129))
+#loc152 = loc(callsite(#loc27 at #loc143))
+#loc154 = loc(callsite(#loc102 at #loc149))
+#loc155 = loc(callsite(#loc103 at #loc149))
+#loc156 = loc(callsite(#loc104 at #loc149))
+#loc158 = loc(callsite(#loc106 at #loc149))
+#loc159 = loc(callsite(#loc107 at #loc149))
+#loc160 = loc(callsite(#loc108 at #loc149))
+#loc162 = loc(callsite(#loc110 at #loc149))
+#loc163 = loc(callsite(#loc111 at #loc149))
+#loc164 = loc(callsite(#loc112 at #loc149))
+#loc165 = loc(callsite(#loc113 at #loc149))
+#loc166 = loc(callsite(#loc114 at #loc149))
+#loc167 = loc(callsite(#loc115 at #loc149))
+#loc168 = loc(callsite(#loc116 at #loc149))
+#loc170 = loc(callsite(#loc119 at #loc149))
+#loc171 = loc(callsite(#loc120 at #loc149))
+#loc172 = loc(callsite(#loc121 at #loc149))
+#loc173 = loc(callsite(#loc122 at #loc149))
+#loc175 = loc(callsite(#loc124 at #loc149))
+#loc176 = loc(callsite(#loc125 at #loc149))
+#loc177 = loc(callsite(#loc126 at #loc149))
+#loc178 = loc(callsite(#loc127 at #loc149))
+#loc179 = loc(callsite(#loc150 at #loc149))
+#loc180 = loc(callsite(#loc151 at #loc149))
+#loc181 = loc(callsite(#loc130 at #loc149))
+#loc182 = loc(callsite(#loc131 at #loc149))
+#loc183 = loc(callsite(#loc132 at #loc149))
+#loc184 = loc(callsite(#loc133 at #loc149))
+#loc185 = loc(callsite(#loc134 at #loc149))
+#loc186 = loc(callsite(#loc135 at #loc149))
+#loc187 = loc(callsite(#loc136 at #loc149))
+#loc188 = loc(callsite(#loc137 at #loc149))
+#loc189 = loc(callsite(#loc138 at #loc149))
+#loc190 = loc(callsite(#loc139 at #loc149))
+#loc191 = loc(callsite(#loc140 at #loc149))
+#loc192 = loc(callsite(#loc29 at #loc152))
+#loc193 = loc(callsite(#loc27 at #loc157))
+#loc195 = loc(callsite(#loc27 at #loc161))
+#loc197 = loc(callsite(#loc117 at #loc169))
+#loc198 = loc(callsite(#loc27 at #loc169))
+#loc200 = loc(callsite(#loc117 at #loc174))
+#loc201 = loc(callsite(#loc27 at #loc174))
+#loc203 = loc(callsite(#loc29 at #loc193))
+#loc204 = loc(callsite(#loc29 at #loc195))
+#loc205 = loc(callsite(#loc29 at #loc198))
+#loc206 = loc(callsite(#loc29 at #loc201))

SpecForge-ext/cache/compiled_kernels/triton/0/4UGGVGDRILN3TLIINH7RHS2I4LFNQRBLGHV63VTW6KCPABX4L4FA/triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3.ttir

@@ -0,0 +1,799 @@
+#loc = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":18:0)
+#loc1 = loc(unknown)
+#loc2 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":41:67)
+#loc23 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":662:12)
+#loc28 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":634:73)
+#loc32 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:51)
+#loc37 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:53)
+#loc46 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:50)
+#loc51 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:51)
+#loc70 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":45:26)
+#loc80 = loc("in_ptr0"(#loc))
+#loc81 = loc("out_ptr2"(#loc))
+#loc82 = loc("out_ptr3"(#loc))
+#loc83 = loc("xnumel"(#loc))
+#loc84 = loc("r0_numel"(#loc))
+#loc106 = loc(callsite(#loc23 at #loc2))
+#loc113 = loc("ileft"(#loc32))
+#loc117 = loc("iright"(#loc37))
+#loc126 = loc("left_idx"(#loc46))
+#loc131 = loc("right_idx"(#loc51))
+#loc150 = loc("tmp11"(#loc70))
+#loc157 = loc(callsite(#loc28 at #loc106))
+#loc161 = loc(callsite(#loc1 at #loc150))
+#loc165 = loc(callsite(#loc113 at #loc157))
+#loc169 = loc(callsite(#loc117 at #loc157))
+#loc177 = loc(callsite(#loc126 at #loc157))
+#loc182 = loc(callsite(#loc131 at #loc157))
+#loc202 = loc(callsite(#loc1 at #loc165))
+#loc204 = loc(callsite(#loc1 at #loc169))
+#loc207 = loc(callsite(#loc1 at #loc177))
+#loc210 = loc(callsite(#loc1 at #loc182))
+module {
+  tt.func public @triton_per_fused__to_copy_clone_slice_sort_sum_transpose_3(%in_ptr0: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("in_ptr0"(#loc)), %out_ptr2: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr2"(#loc)), %out_ptr3: !tt.ptr<i32> {tt.divisibility = 16 : i32} loc("out_ptr3"(#loc)), %xnumel: i32 {tt.divisibility = 16 : i32} loc("xnumel"(#loc)), %r0_numel: i32 {tt.divisibility = 16 : i32} loc("r0_numel"(#loc))) attributes {noinline = false} {
+    %cst = arith.constant dense<1> : tensor<1x2x1xi32> loc(#loc85)
+    %cst_0 = arith.constant dense<0> : tensor<32x16xi32> loc(#loc1)
+    %tmp10 = arith.constant dense<0> : tensor<32x16xi64> loc(#loc86)
+    %tmp0 = arith.constant dense<272> : tensor<32x1xi32> loc(#loc87)
+    %tmp0_1 = arith.constant dense<17> : tensor<1x16xi32> loc(#loc88)
+    %cst_2 = arith.constant dense<16> : tensor<32x1xi32> loc(#loc1)
+    %xmask = arith.constant dense<128> : tensor<32x1xi32> loc(#loc89)
+    %c32_i32 = arith.constant 32 : i32 loc(#loc1)
+    %xoffset = tt.get_program_id x : i32 loc(#loc90)
+    %xoffset_3 = arith.muli %xoffset, %c32_i32 : i32 loc(#loc91)
+    %xindex = tt.make_range {end = 32 : i32, start = 0 : i32} : tensor<32xi32> loc(#loc92)
+    %xindex_4 = tt.expand_dims %xindex {axis = 1 : i32} : tensor<32xi32> -> tensor<32x1xi32> loc(#loc93)
+    %xindex_5 = tt.splat %xoffset_3 : i32 -> tensor<32x1xi32> loc(#loc94)
+    %xindex_6 = arith.addi %xindex_5, %xindex_4 : tensor<32x1xi32> loc(#loc94)
+    %xmask_7 = arith.cmpi slt, %xindex_6, %xmask : tensor<32x1xi32> loc(#loc89)
+    %r0_index = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32> loc(#loc95)
+    %r0_index_8 = tt.expand_dims %r0_index {axis = 0 : i32} : tensor<16xi32> -> tensor<1x16xi32> loc(#loc96)
+    %x0 = arith.remsi %xindex_6, %cst_2 : tensor<32x1xi32> loc(#loc97)
+    %x1 = arith.divsi %xindex_6, %cst_2 : tensor<32x1xi32> loc(#loc98)
+    %tmp0_9 = arith.muli %r0_index_8, %tmp0_1 : tensor<1x16xi32> loc(#loc88)
+    %tmp0_10 = tt.broadcast %x0 : tensor<32x1xi32> -> tensor<32x16xi32> loc(#loc99)
+    %tmp0_11 = tt.broadcast %tmp0_9 : tensor<1x16xi32> -> tensor<32x16xi32> loc(#loc99)
+    %tmp0_12 = arith.addi %tmp0_10, %tmp0_11 : tensor<32x16xi32> loc(#loc99)
+    %tmp0_13 = arith.muli %x1, %tmp0 : tensor<32x1xi32> loc(#loc87)
+    %tmp0_14 = tt.broadcast %tmp0_13 : tensor<32x1xi32> -> tensor<32x16xi32> loc(#loc100)
+    %tmp0_15 = arith.addi %tmp0_12, %tmp0_14 : tensor<32x16xi32> loc(#loc100)
+    %tmp0_16 = tt.splat %in_ptr0 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>> loc(#loc101)
+    %tmp0_17 = tt.addptr %tmp0_16, %tmp0_15 : tensor<32x16x!tt.ptr<i32>>, tensor<32x16xi32> loc(#loc101)
+    %tmp0_18 = tt.broadcast %xmask_7 : tensor<32x1xi1> -> tensor<32x16xi1> loc(#loc102)
+    %tmp0_19 = tt.load %tmp0_17, %tmp0_18, %cst_0 : tensor<32x16x!tt.ptr<i32>> loc(#loc102)
+    %tmp2 = arith.trunci %r0_index_8 : tensor<1x16xi32> to tensor<1x16xi16> loc(#loc103)
+    %tmp4 = tt.broadcast %tmp2 : tensor<1x16xi16> -> tensor<32x16xi16> loc(#loc104)
+    %flip = tt.make_range {end = 2 : i32, start = 0 : i32} : tensor<2xi32> loc(#loc153)
+    %flip_20 = tt.expand_dims %flip {axis = 0 : i32} : tensor<2xi32> -> tensor<1x2xi32> loc(#loc154)
+    %flip_21 = tt.expand_dims %flip_20 {axis = 2 : i32} : tensor<1x2xi32> -> tensor<1x2x1xi32> loc(#loc154)
+    %flip_22 = tt.broadcast %flip_21 : tensor<1x2x1xi32> -> tensor<128x2x2xi32> loc(#loc155)
+    %flip_23 = tt.reshape %flip_22 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc156)
+    %y = tt.reshape %tmp0_19 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc162)
+    %left_mask = arith.subi %cst, %flip_21 : tensor<1x2x1xi32> loc(#loc163)
+    %ileft = tt.broadcast %left_mask : tensor<1x2x1xi32> -> tensor<256x2x1xi32> loc(#loc164)
+    %ileft_24 = arith.muli %y, %ileft : tensor<256x2x1xi32> loc(#loc164)
+    %ileft_25 = "tt.reduce"(%ileft_24) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc201)
+    %ileft_26 = tt.expand_dims %ileft_25 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc166)
+    %ileft_27 = tt.broadcast %ileft_26 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc167)
+    %iright = tt.broadcast %flip_21 : tensor<1x2x1xi32> -> tensor<256x2x1xi32> loc(#loc168)
+    %iright_28 = arith.muli %y, %iright : tensor<256x2x1xi32> loc(#loc168)
+    %iright_29 = "tt.reduce"(%iright_28) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc203)
+    %iright_30 = tt.expand_dims %iright_29 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc170)
+    %iright_31 = tt.broadcast %iright_30 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc171)
+    %ileft_32 = tt.reshape %ileft_27 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_33 = tt.reshape %iright_31 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx = tt.reshape %tmp4 : tensor<32x16xi16> -> tensor<256x2x1xi16> loc(#loc174)
+    %left_idx = arith.trunci %left_mask : tensor<1x2x1xi32> to tensor<1x2x1xi16> loc(#loc175)
+    %left_idx_34 = tt.broadcast %left_idx : tensor<1x2x1xi16> -> tensor<256x2x1xi16> loc(#loc176)
+    %left_idx_35 = arith.muli %y_idx, %left_idx_34 : tensor<256x2x1xi16> loc(#loc176)
+    %input = arith.extsi %left_idx_35 : tensor<256x2x1xi16> to tensor<256x2x1xi32> loc(#loc205)
+    %left_idx_36 = "tt.reduce"(%input) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc206)
+    %left_idx_37 = tt.expand_dims %left_idx_36 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc178)
+    %left_idx_38 = tt.broadcast %left_idx_37 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc179)
+    %right_idx = arith.trunci %flip_21 : tensor<1x2x1xi32> to tensor<1x2x1xi16> loc(#loc180)
+    %right_idx_39 = tt.broadcast %right_idx : tensor<1x2x1xi16> -> tensor<256x2x1xi16> loc(#loc181)
+    %right_idx_40 = arith.muli %y_idx, %right_idx_39 : tensor<256x2x1xi16> loc(#loc181)
+    %input_41 = arith.extsi %right_idx_40 : tensor<256x2x1xi16> to tensor<256x2x1xi32> loc(#loc208)
+    %right_idx_42 = "tt.reduce"(%input_41) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc209)
+    %right_idx_43 = tt.expand_dims %right_idx_42 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc183)
+    %right_idx_44 = tt.broadcast %right_idx_43 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc184)
+    %left_idx_45 = tt.reshape %left_idx_38 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_46 = tt.reshape %right_idx_44 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond = arith.cmpi slt, %ileft_32, %iright_33 : tensor<32x16xi32> loc(#loc187)
+    %eq = arith.cmpi eq, %ileft_32, %iright_33 : tensor<32x16xi32> loc(#loc188)
+    %cond_47 = arith.cmpi sgt, %left_idx_45, %right_idx_46 : tensor<32x16xi32> loc(#loc189)
+    %cond_48 = arith.andi %eq, %cond_47 : tensor<32x16xi1> loc(#loc190)
+    %cond_49 = arith.ori %cond, %cond_48 : tensor<32x16xi1> loc(#loc191)
+    %cond_50 = arith.extui %cond_49 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_51 = arith.xori %cond_50, %flip_23 : tensor<32x16xi32> loc(#loc192)
+    %cond_52 = arith.cmpi ne, %cond_51, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret = arith.xori %ileft_32, %iright_33 : tensor<32x16xi32> loc(#loc194)
+    %ret_53 = arith.select %cond_52, %ret, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_54 = arith.xori %tmp0_19, %ret_53 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs = arith.xori %left_idx_45, %right_idx_46 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_55 = arith.select %cond_52, %new_idxs, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_56 = arith.extsi %tmp2 : tensor<1x16xi16> to tensor<1x16xi32> loc(#loc199)
+    %new_idxs_57 = tt.broadcast %new_idxs_56 : tensor<1x16xi32> -> tensor<32x16xi32> loc(#loc199)
+    %new_idxs_58 = arith.xori %new_idxs_57, %new_idxs_55 : tensor<32x16xi32> loc(#loc199)
+    %flip_59 = tt.broadcast %flip_21 : tensor<1x2x1xi32> -> tensor<64x2x4xi32> loc(#loc155)
+    %flip_60 = tt.reshape %flip_59 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc156)
+    %y_61 = tt.reshape %ret_54 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc162)
+    %ileft_62 = tt.broadcast %left_mask : tensor<1x2x1xi32> -> tensor<128x2x2xi32> loc(#loc164)
+    %ileft_63 = arith.muli %y_61, %ileft_62 : tensor<128x2x2xi32> loc(#loc164)
+    %ileft_64 = "tt.reduce"(%ileft_63) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc201)
+    %ileft_65 = tt.expand_dims %ileft_64 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc166)
+    %ileft_66 = tt.broadcast %ileft_65 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc167)
+    %iright_67 = arith.muli %y_61, %flip_22 : tensor<128x2x2xi32> loc(#loc168)
+    %iright_68 = "tt.reduce"(%iright_67) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc203)
+    %iright_69 = tt.expand_dims %iright_68 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc170)
+    %iright_70 = tt.broadcast %iright_69 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc171)
+    %ileft_71 = tt.reshape %ileft_66 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_72 = tt.reshape %iright_70 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_73 = tt.reshape %new_idxs_58 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc174)
+    %left_idx_74 = arith.muli %y_idx_73, %ileft_62 : tensor<128x2x2xi32> loc(#loc176)
+    %left_idx_75 = "tt.reduce"(%left_idx_74) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc206)
+    %left_idx_76 = tt.expand_dims %left_idx_75 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc178)
+    %left_idx_77 = tt.broadcast %left_idx_76 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc179)
+    %right_idx_78 = arith.muli %y_idx_73, %flip_22 : tensor<128x2x2xi32> loc(#loc181)
+    %right_idx_79 = "tt.reduce"(%right_idx_78) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc209)
+    %right_idx_80 = tt.expand_dims %right_idx_79 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc183)
+    %right_idx_81 = tt.broadcast %right_idx_80 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc184)
+    %left_idx_82 = tt.reshape %left_idx_77 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_83 = tt.reshape %right_idx_81 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_84 = arith.cmpi slt, %ileft_71, %iright_72 : tensor<32x16xi32> loc(#loc187)
+    %eq_85 = arith.cmpi eq, %ileft_71, %iright_72 : tensor<32x16xi32> loc(#loc188)
+    %cond_86 = arith.cmpi sgt, %left_idx_82, %right_idx_83 : tensor<32x16xi32> loc(#loc189)
+    %cond_87 = arith.andi %eq_85, %cond_86 : tensor<32x16xi1> loc(#loc190)
+    %cond_88 = arith.ori %cond_84, %cond_87 : tensor<32x16xi1> loc(#loc191)
+    %cond_89 = arith.extui %cond_88 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_90 = arith.xori %cond_89, %flip_60 : tensor<32x16xi32> loc(#loc192)
+    %cond_91 = arith.cmpi ne, %cond_90, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret_92 = arith.xori %ileft_71, %iright_72 : tensor<32x16xi32> loc(#loc194)
+    %ret_93 = arith.select %cond_91, %ret_92, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_94 = arith.xori %ret_54, %ret_93 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_95 = arith.xori %left_idx_82, %right_idx_83 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_96 = arith.select %cond_91, %new_idxs_95, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_97 = arith.xori %new_idxs_58, %new_idxs_96 : tensor<32x16xi32> loc(#loc199)
+    %y_98 = tt.reshape %ret_94 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc162)
+    %ileft_99 = arith.muli %y_98, %ileft : tensor<256x2x1xi32> loc(#loc164)
+    %ileft_100 = "tt.reduce"(%ileft_99) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc201)
+    %ileft_101 = tt.expand_dims %ileft_100 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc166)
+    %ileft_102 = tt.broadcast %ileft_101 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc167)
+    %iright_103 = arith.muli %y_98, %iright : tensor<256x2x1xi32> loc(#loc168)
+    %iright_104 = "tt.reduce"(%iright_103) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc203)
+    %iright_105 = tt.expand_dims %iright_104 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc170)
+    %iright_106 = tt.broadcast %iright_105 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc171)
+    %ileft_107 = tt.reshape %ileft_102 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_108 = tt.reshape %iright_106 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_109 = tt.reshape %new_idxs_97 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc174)
+    %left_idx_110 = arith.muli %y_idx_109, %ileft : tensor<256x2x1xi32> loc(#loc176)
+    %left_idx_111 = "tt.reduce"(%left_idx_110) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc206)
+    %left_idx_112 = tt.expand_dims %left_idx_111 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc178)
+    %left_idx_113 = tt.broadcast %left_idx_112 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc179)
+    %right_idx_114 = arith.muli %y_idx_109, %iright : tensor<256x2x1xi32> loc(#loc181)
+    %right_idx_115 = "tt.reduce"(%right_idx_114) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc209)
+    %right_idx_116 = tt.expand_dims %right_idx_115 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc183)
+    %right_idx_117 = tt.broadcast %right_idx_116 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc184)
+    %left_idx_118 = tt.reshape %left_idx_113 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_119 = tt.reshape %right_idx_117 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_120 = arith.cmpi slt, %ileft_107, %iright_108 : tensor<32x16xi32> loc(#loc187)
+    %eq_121 = arith.cmpi eq, %ileft_107, %iright_108 : tensor<32x16xi32> loc(#loc188)
+    %cond_122 = arith.cmpi sgt, %left_idx_118, %right_idx_119 : tensor<32x16xi32> loc(#loc189)
+    %cond_123 = arith.andi %eq_121, %cond_122 : tensor<32x16xi1> loc(#loc190)
+    %cond_124 = arith.ori %cond_120, %cond_123 : tensor<32x16xi1> loc(#loc191)
+    %cond_125 = arith.extui %cond_124 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_126 = arith.xori %cond_125, %flip_60 : tensor<32x16xi32> loc(#loc192)
+    %cond_127 = arith.cmpi ne, %cond_126, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret_128 = arith.xori %ileft_107, %iright_108 : tensor<32x16xi32> loc(#loc194)
+    %ret_129 = arith.select %cond_127, %ret_128, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_130 = arith.xori %ret_94, %ret_129 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_131 = arith.xori %left_idx_118, %right_idx_119 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_132 = arith.select %cond_127, %new_idxs_131, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_133 = arith.xori %new_idxs_97, %new_idxs_132 : tensor<32x16xi32> loc(#loc199)
+    %flip_134 = tt.broadcast %flip_21 : tensor<1x2x1xi32> -> tensor<32x2x8xi32> loc(#loc155)
+    %flip_135 = tt.reshape %flip_134 : tensor<32x2x8xi32> -> tensor<32x16xi32> loc(#loc156)
+    %y_136 = tt.reshape %ret_130 : tensor<32x16xi32> -> tensor<64x2x4xi32> loc(#loc162)
+    %ileft_137 = tt.broadcast %left_mask : tensor<1x2x1xi32> -> tensor<64x2x4xi32> loc(#loc164)
+    %ileft_138 = arith.muli %y_136, %ileft_137 : tensor<64x2x4xi32> loc(#loc164)
+    %ileft_139 = "tt.reduce"(%ileft_138) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc201)
+    %ileft_140 = tt.expand_dims %ileft_139 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc166)
+    %ileft_141 = tt.broadcast %ileft_140 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc167)
+    %iright_142 = arith.muli %y_136, %flip_59 : tensor<64x2x4xi32> loc(#loc168)
+    %iright_143 = "tt.reduce"(%iright_142) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc203)
+    %iright_144 = tt.expand_dims %iright_143 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc170)
+    %iright_145 = tt.broadcast %iright_144 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc171)
+    %ileft_146 = tt.reshape %ileft_141 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_147 = tt.reshape %iright_145 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_148 = tt.reshape %new_idxs_133 : tensor<32x16xi32> -> tensor<64x2x4xi32> loc(#loc174)
+    %left_idx_149 = arith.muli %y_idx_148, %ileft_137 : tensor<64x2x4xi32> loc(#loc176)
+    %left_idx_150 = "tt.reduce"(%left_idx_149) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc206)
+    %left_idx_151 = tt.expand_dims %left_idx_150 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc178)
+    %left_idx_152 = tt.broadcast %left_idx_151 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc179)
+    %right_idx_153 = arith.muli %y_idx_148, %flip_59 : tensor<64x2x4xi32> loc(#loc181)
+    %right_idx_154 = "tt.reduce"(%right_idx_153) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc209)
+    %right_idx_155 = tt.expand_dims %right_idx_154 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc183)
+    %right_idx_156 = tt.broadcast %right_idx_155 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc184)
+    %left_idx_157 = tt.reshape %left_idx_152 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_158 = tt.reshape %right_idx_156 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_159 = arith.cmpi slt, %ileft_146, %iright_147 : tensor<32x16xi32> loc(#loc187)
+    %eq_160 = arith.cmpi eq, %ileft_146, %iright_147 : tensor<32x16xi32> loc(#loc188)
+    %cond_161 = arith.cmpi sgt, %left_idx_157, %right_idx_158 : tensor<32x16xi32> loc(#loc189)
+    %cond_162 = arith.andi %eq_160, %cond_161 : tensor<32x16xi1> loc(#loc190)
+    %cond_163 = arith.ori %cond_159, %cond_162 : tensor<32x16xi1> loc(#loc191)
+    %cond_164 = arith.extui %cond_163 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_165 = arith.xori %cond_164, %flip_135 : tensor<32x16xi32> loc(#loc192)
+    %cond_166 = arith.cmpi ne, %cond_165, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret_167 = arith.xori %ileft_146, %iright_147 : tensor<32x16xi32> loc(#loc194)
+    %ret_168 = arith.select %cond_166, %ret_167, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_169 = arith.xori %ret_130, %ret_168 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_170 = arith.xori %left_idx_157, %right_idx_158 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_171 = arith.select %cond_166, %new_idxs_170, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_172 = arith.xori %new_idxs_133, %new_idxs_171 : tensor<32x16xi32> loc(#loc199)
+    %y_173 = tt.reshape %ret_169 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc162)
+    %ileft_174 = arith.muli %y_173, %ileft_62 : tensor<128x2x2xi32> loc(#loc164)
+    %ileft_175 = "tt.reduce"(%ileft_174) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc201)
+    %ileft_176 = tt.expand_dims %ileft_175 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc166)
+    %ileft_177 = tt.broadcast %ileft_176 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc167)
+    %iright_178 = arith.muli %y_173, %flip_22 : tensor<128x2x2xi32> loc(#loc168)
+    %iright_179 = "tt.reduce"(%iright_178) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc203)
+    %iright_180 = tt.expand_dims %iright_179 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc170)
+    %iright_181 = tt.broadcast %iright_180 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc171)
+    %ileft_182 = tt.reshape %ileft_177 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_183 = tt.reshape %iright_181 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_184 = tt.reshape %new_idxs_172 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc174)
+    %left_idx_185 = arith.muli %y_idx_184, %ileft_62 : tensor<128x2x2xi32> loc(#loc176)
+    %left_idx_186 = "tt.reduce"(%left_idx_185) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc206)
+    %left_idx_187 = tt.expand_dims %left_idx_186 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc178)
+    %left_idx_188 = tt.broadcast %left_idx_187 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc179)
+    %right_idx_189 = arith.muli %y_idx_184, %flip_22 : tensor<128x2x2xi32> loc(#loc181)
+    %right_idx_190 = "tt.reduce"(%right_idx_189) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc209)
+    %right_idx_191 = tt.expand_dims %right_idx_190 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc183)
+    %right_idx_192 = tt.broadcast %right_idx_191 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc184)
+    %left_idx_193 = tt.reshape %left_idx_188 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_194 = tt.reshape %right_idx_192 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_195 = arith.cmpi slt, %ileft_182, %iright_183 : tensor<32x16xi32> loc(#loc187)
+    %eq_196 = arith.cmpi eq, %ileft_182, %iright_183 : tensor<32x16xi32> loc(#loc188)
+    %cond_197 = arith.cmpi sgt, %left_idx_193, %right_idx_194 : tensor<32x16xi32> loc(#loc189)
+    %cond_198 = arith.andi %eq_196, %cond_197 : tensor<32x16xi1> loc(#loc190)
+    %cond_199 = arith.ori %cond_195, %cond_198 : tensor<32x16xi1> loc(#loc191)
+    %cond_200 = arith.extui %cond_199 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_201 = arith.xori %cond_200, %flip_135 : tensor<32x16xi32> loc(#loc192)
+    %cond_202 = arith.cmpi ne, %cond_201, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret_203 = arith.xori %ileft_182, %iright_183 : tensor<32x16xi32> loc(#loc194)
+    %ret_204 = arith.select %cond_202, %ret_203, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_205 = arith.xori %ret_169, %ret_204 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_206 = arith.xori %left_idx_193, %right_idx_194 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_207 = arith.select %cond_202, %new_idxs_206, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_208 = arith.xori %new_idxs_172, %new_idxs_207 : tensor<32x16xi32> loc(#loc199)
+    %y_209 = tt.reshape %ret_205 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc162)
+    %ileft_210 = arith.muli %y_209, %ileft : tensor<256x2x1xi32> loc(#loc164)
+    %ileft_211 = "tt.reduce"(%ileft_210) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc201)
+    %ileft_212 = tt.expand_dims %ileft_211 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc166)
+    %ileft_213 = tt.broadcast %ileft_212 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc167)
+    %iright_214 = arith.muli %y_209, %iright : tensor<256x2x1xi32> loc(#loc168)
+    %iright_215 = "tt.reduce"(%iright_214) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc203)
+    %iright_216 = tt.expand_dims %iright_215 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc170)
+    %iright_217 = tt.broadcast %iright_216 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc171)
+    %ileft_218 = tt.reshape %ileft_213 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_219 = tt.reshape %iright_217 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_220 = tt.reshape %new_idxs_208 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc174)
+    %left_idx_221 = arith.muli %y_idx_220, %ileft : tensor<256x2x1xi32> loc(#loc176)
+    %left_idx_222 = "tt.reduce"(%left_idx_221) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc206)
+    %left_idx_223 = tt.expand_dims %left_idx_222 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc178)
+    %left_idx_224 = tt.broadcast %left_idx_223 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc179)
+    %right_idx_225 = arith.muli %y_idx_220, %iright : tensor<256x2x1xi32> loc(#loc181)
+    %right_idx_226 = "tt.reduce"(%right_idx_225) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc209)
+    %right_idx_227 = tt.expand_dims %right_idx_226 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc183)
+    %right_idx_228 = tt.broadcast %right_idx_227 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc184)
+    %left_idx_229 = tt.reshape %left_idx_224 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_230 = tt.reshape %right_idx_228 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_231 = arith.cmpi slt, %ileft_218, %iright_219 : tensor<32x16xi32> loc(#loc187)
+    %eq_232 = arith.cmpi eq, %ileft_218, %iright_219 : tensor<32x16xi32> loc(#loc188)
+    %cond_233 = arith.cmpi sgt, %left_idx_229, %right_idx_230 : tensor<32x16xi32> loc(#loc189)
+    %cond_234 = arith.andi %eq_232, %cond_233 : tensor<32x16xi1> loc(#loc190)
+    %cond_235 = arith.ori %cond_231, %cond_234 : tensor<32x16xi1> loc(#loc191)
+    %cond_236 = arith.extui %cond_235 : tensor<32x16xi1> to tensor<32x16xi32> loc(#loc192)
+    %cond_237 = arith.xori %cond_236, %flip_135 : tensor<32x16xi32> loc(#loc192)
+    %cond_238 = arith.cmpi ne, %cond_237, %cst_0 : tensor<32x16xi32> loc(#loc193)
+    %ret_239 = arith.xori %ileft_218, %iright_219 : tensor<32x16xi32> loc(#loc194)
+    %ret_240 = arith.select %cond_238, %ret_239, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_241 = arith.xori %ret_205, %ret_240 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_242 = arith.xori %left_idx_229, %right_idx_230 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_243 = arith.select %cond_238, %new_idxs_242, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_244 = arith.xori %new_idxs_208, %new_idxs_243 : tensor<32x16xi32> loc(#loc199)
+    %y_245 = tt.reshape %ret_241 : tensor<32x16xi32> -> tensor<32x2x8xi32> loc(#loc162)
+    %ileft_246 = tt.broadcast %left_mask : tensor<1x2x1xi32> -> tensor<32x2x8xi32> loc(#loc164)
+    %ileft_247 = arith.muli %y_245, %ileft_246 : tensor<32x2x8xi32> loc(#loc164)
+    %ileft_248 = "tt.reduce"(%ileft_247) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<32x2x8xi32>) -> tensor<32x8xi32> loc(#loc201)
+    %ileft_249 = tt.expand_dims %ileft_248 {axis = 1 : i32} : tensor<32x8xi32> -> tensor<32x1x8xi32> loc(#loc166)
+    %ileft_250 = tt.broadcast %ileft_249 : tensor<32x1x8xi32> -> tensor<32x2x8xi32> loc(#loc167)
+    %iright_251 = arith.muli %y_245, %flip_134 : tensor<32x2x8xi32> loc(#loc168)
+    %iright_252 = "tt.reduce"(%iright_251) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<32x2x8xi32>) -> tensor<32x8xi32> loc(#loc203)
+    %iright_253 = tt.expand_dims %iright_252 {axis = 1 : i32} : tensor<32x8xi32> -> tensor<32x1x8xi32> loc(#loc170)
+    %iright_254 = tt.broadcast %iright_253 : tensor<32x1x8xi32> -> tensor<32x2x8xi32> loc(#loc171)
+    %ileft_255 = tt.reshape %ileft_250 : tensor<32x2x8xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_256 = tt.reshape %iright_254 : tensor<32x2x8xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_257 = tt.reshape %new_idxs_244 : tensor<32x16xi32> -> tensor<32x2x8xi32> loc(#loc174)
+    %left_idx_258 = arith.muli %y_idx_257, %ileft_246 : tensor<32x2x8xi32> loc(#loc176)
+    %left_idx_259 = "tt.reduce"(%left_idx_258) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<32x2x8xi32>) -> tensor<32x8xi32> loc(#loc206)
+    %left_idx_260 = tt.expand_dims %left_idx_259 {axis = 1 : i32} : tensor<32x8xi32> -> tensor<32x1x8xi32> loc(#loc178)
+    %left_idx_261 = tt.broadcast %left_idx_260 : tensor<32x1x8xi32> -> tensor<32x2x8xi32> loc(#loc179)
+    %right_idx_262 = arith.muli %y_idx_257, %flip_134 : tensor<32x2x8xi32> loc(#loc181)
+    %right_idx_263 = "tt.reduce"(%right_idx_262) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<32x2x8xi32>) -> tensor<32x8xi32> loc(#loc209)
+    %right_idx_264 = tt.expand_dims %right_idx_263 {axis = 1 : i32} : tensor<32x8xi32> -> tensor<32x1x8xi32> loc(#loc183)
+    %right_idx_265 = tt.broadcast %right_idx_264 : tensor<32x1x8xi32> -> tensor<32x2x8xi32> loc(#loc184)
+    %left_idx_266 = tt.reshape %left_idx_261 : tensor<32x2x8xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_267 = tt.reshape %right_idx_265 : tensor<32x2x8xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_268 = arith.cmpi slt, %ileft_255, %iright_256 : tensor<32x16xi32> loc(#loc187)
+    %eq_269 = arith.cmpi eq, %ileft_255, %iright_256 : tensor<32x16xi32> loc(#loc188)
+    %cond_270 = arith.cmpi sgt, %left_idx_266, %right_idx_267 : tensor<32x16xi32> loc(#loc189)
+    %cond_271 = arith.andi %eq_269, %cond_270 : tensor<32x16xi1> loc(#loc190)
+    %cond_272 = arith.ori %cond_268, %cond_271 : tensor<32x16xi1> loc(#loc191)
+    %ret_273 = arith.xori %ileft_255, %iright_256 : tensor<32x16xi32> loc(#loc194)
+    %ret_274 = arith.select %cond_272, %ret_273, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_275 = arith.xori %ret_241, %ret_274 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_276 = arith.xori %left_idx_266, %right_idx_267 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_277 = arith.select %cond_272, %new_idxs_276, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_278 = arith.xori %new_idxs_244, %new_idxs_277 : tensor<32x16xi32> loc(#loc199)
+    %y_279 = tt.reshape %ret_275 : tensor<32x16xi32> -> tensor<64x2x4xi32> loc(#loc162)
+    %ileft_280 = arith.muli %y_279, %ileft_137 : tensor<64x2x4xi32> loc(#loc164)
+    %ileft_281 = "tt.reduce"(%ileft_280) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc201)
+    %ileft_282 = tt.expand_dims %ileft_281 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc166)
+    %ileft_283 = tt.broadcast %ileft_282 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc167)
+    %iright_284 = arith.muli %y_279, %flip_59 : tensor<64x2x4xi32> loc(#loc168)
+    %iright_285 = "tt.reduce"(%iright_284) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc203)
+    %iright_286 = tt.expand_dims %iright_285 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc170)
+    %iright_287 = tt.broadcast %iright_286 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc171)
+    %ileft_288 = tt.reshape %ileft_283 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_289 = tt.reshape %iright_287 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_290 = tt.reshape %new_idxs_278 : tensor<32x16xi32> -> tensor<64x2x4xi32> loc(#loc174)
+    %left_idx_291 = arith.muli %y_idx_290, %ileft_137 : tensor<64x2x4xi32> loc(#loc176)
+    %left_idx_292 = "tt.reduce"(%left_idx_291) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc206)
+    %left_idx_293 = tt.expand_dims %left_idx_292 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc178)
+    %left_idx_294 = tt.broadcast %left_idx_293 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc179)
+    %right_idx_295 = arith.muli %y_idx_290, %flip_59 : tensor<64x2x4xi32> loc(#loc181)
+    %right_idx_296 = "tt.reduce"(%right_idx_295) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<64x2x4xi32>) -> tensor<64x4xi32> loc(#loc209)
+    %right_idx_297 = tt.expand_dims %right_idx_296 {axis = 1 : i32} : tensor<64x4xi32> -> tensor<64x1x4xi32> loc(#loc183)
+    %right_idx_298 = tt.broadcast %right_idx_297 : tensor<64x1x4xi32> -> tensor<64x2x4xi32> loc(#loc184)
+    %left_idx_299 = tt.reshape %left_idx_294 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_300 = tt.reshape %right_idx_298 : tensor<64x2x4xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_301 = arith.cmpi slt, %ileft_288, %iright_289 : tensor<32x16xi32> loc(#loc187)
+    %eq_302 = arith.cmpi eq, %ileft_288, %iright_289 : tensor<32x16xi32> loc(#loc188)
+    %cond_303 = arith.cmpi sgt, %left_idx_299, %right_idx_300 : tensor<32x16xi32> loc(#loc189)
+    %cond_304 = arith.andi %eq_302, %cond_303 : tensor<32x16xi1> loc(#loc190)
+    %cond_305 = arith.ori %cond_301, %cond_304 : tensor<32x16xi1> loc(#loc191)
+    %ret_306 = arith.xori %ileft_288, %iright_289 : tensor<32x16xi32> loc(#loc194)
+    %ret_307 = arith.select %cond_305, %ret_306, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_308 = arith.xori %ret_275, %ret_307 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_309 = arith.xori %left_idx_299, %right_idx_300 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_310 = arith.select %cond_305, %new_idxs_309, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_311 = arith.xori %new_idxs_278, %new_idxs_310 : tensor<32x16xi32> loc(#loc199)
+    %y_312 = tt.reshape %ret_308 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc162)
+    %ileft_313 = arith.muli %y_312, %ileft_62 : tensor<128x2x2xi32> loc(#loc164)
+    %ileft_314 = "tt.reduce"(%ileft_313) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc201)
+    %ileft_315 = tt.expand_dims %ileft_314 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc166)
+    %ileft_316 = tt.broadcast %ileft_315 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc167)
+    %iright_317 = arith.muli %y_312, %flip_22 : tensor<128x2x2xi32> loc(#loc168)
+    %iright_318 = "tt.reduce"(%iright_317) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc203)
+    %iright_319 = tt.expand_dims %iright_318 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc170)
+    %iright_320 = tt.broadcast %iright_319 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc171)
+    %ileft_321 = tt.reshape %ileft_316 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_322 = tt.reshape %iright_320 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_323 = tt.reshape %new_idxs_311 : tensor<32x16xi32> -> tensor<128x2x2xi32> loc(#loc174)
+    %left_idx_324 = arith.muli %y_idx_323, %ileft_62 : tensor<128x2x2xi32> loc(#loc176)
+    %left_idx_325 = "tt.reduce"(%left_idx_324) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc206)
+    %left_idx_326 = tt.expand_dims %left_idx_325 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc178)
+    %left_idx_327 = tt.broadcast %left_idx_326 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc179)
+    %right_idx_328 = arith.muli %y_idx_323, %flip_22 : tensor<128x2x2xi32> loc(#loc181)
+    %right_idx_329 = "tt.reduce"(%right_idx_328) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<128x2x2xi32>) -> tensor<128x2xi32> loc(#loc209)
+    %right_idx_330 = tt.expand_dims %right_idx_329 {axis = 1 : i32} : tensor<128x2xi32> -> tensor<128x1x2xi32> loc(#loc183)
+    %right_idx_331 = tt.broadcast %right_idx_330 : tensor<128x1x2xi32> -> tensor<128x2x2xi32> loc(#loc184)
+    %left_idx_332 = tt.reshape %left_idx_327 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_333 = tt.reshape %right_idx_331 : tensor<128x2x2xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_334 = arith.cmpi slt, %ileft_321, %iright_322 : tensor<32x16xi32> loc(#loc187)
+    %eq_335 = arith.cmpi eq, %ileft_321, %iright_322 : tensor<32x16xi32> loc(#loc188)
+    %cond_336 = arith.cmpi sgt, %left_idx_332, %right_idx_333 : tensor<32x16xi32> loc(#loc189)
+    %cond_337 = arith.andi %eq_335, %cond_336 : tensor<32x16xi1> loc(#loc190)
+    %cond_338 = arith.ori %cond_334, %cond_337 : tensor<32x16xi1> loc(#loc191)
+    %ret_339 = arith.xori %ileft_321, %iright_322 : tensor<32x16xi32> loc(#loc194)
+    %ret_340 = arith.select %cond_338, %ret_339, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc195)
+    %ret_341 = arith.xori %ret_308, %ret_340 : tensor<32x16xi32> loc(#loc196)
+    %new_idxs_342 = arith.xori %left_idx_332, %right_idx_333 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_343 = arith.select %cond_338, %new_idxs_342, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_344 = arith.xori %new_idxs_311, %new_idxs_343 : tensor<32x16xi32> loc(#loc199)
+    %y_345 = tt.reshape %ret_341 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc162)
+    %ileft_346 = arith.muli %y_345, %ileft : tensor<256x2x1xi32> loc(#loc164)
+    %ileft_347 = "tt.reduce"(%ileft_346) <{axis = 1 : i32}> ({
+    ^bb0(%ileft_377: i32 loc(callsite(#loc1 at #loc165)), %ileft_378: i32 loc(callsite(#loc1 at #loc165))):
+      %ileft_379 = arith.addi %ileft_377, %ileft_378 : i32 loc(#loc211)
+      tt.reduce.return %ileft_379 : i32 loc(#loc201)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc201)
+    %ileft_348 = tt.expand_dims %ileft_347 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc166)
+    %ileft_349 = tt.broadcast %ileft_348 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc167)
+    %iright_350 = arith.muli %y_345, %iright : tensor<256x2x1xi32> loc(#loc168)
+    %iright_351 = "tt.reduce"(%iright_350) <{axis = 1 : i32}> ({
+    ^bb0(%iright_377: i32 loc(callsite(#loc1 at #loc169)), %iright_378: i32 loc(callsite(#loc1 at #loc169))):
+      %iright_379 = arith.addi %iright_377, %iright_378 : i32 loc(#loc212)
+      tt.reduce.return %iright_379 : i32 loc(#loc203)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc203)
+    %iright_352 = tt.expand_dims %iright_351 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc170)
+    %iright_353 = tt.broadcast %iright_352 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc171)
+    %ileft_354 = tt.reshape %ileft_349 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc172)
+    %iright_355 = tt.reshape %iright_353 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc173)
+    %y_idx_356 = tt.reshape %new_idxs_344 : tensor<32x16xi32> -> tensor<256x2x1xi32> loc(#loc174)
+    %left_idx_357 = arith.muli %y_idx_356, %ileft : tensor<256x2x1xi32> loc(#loc176)
+    %left_idx_358 = "tt.reduce"(%left_idx_357) <{axis = 1 : i32}> ({
+    ^bb0(%left_idx_377: i32 loc(callsite(#loc1 at #loc177)), %left_idx_378: i32 loc(callsite(#loc1 at #loc177))):
+      %left_idx_379 = arith.addi %left_idx_377, %left_idx_378 : i32 loc(#loc213)
+      tt.reduce.return %left_idx_379 : i32 loc(#loc206)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc206)
+    %left_idx_359 = tt.expand_dims %left_idx_358 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc178)
+    %left_idx_360 = tt.broadcast %left_idx_359 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc179)
+    %right_idx_361 = arith.muli %y_idx_356, %iright : tensor<256x2x1xi32> loc(#loc181)
+    %right_idx_362 = "tt.reduce"(%right_idx_361) <{axis = 1 : i32}> ({
+    ^bb0(%right_idx_377: i32 loc(callsite(#loc1 at #loc182)), %right_idx_378: i32 loc(callsite(#loc1 at #loc182))):
+      %right_idx_379 = arith.addi %right_idx_377, %right_idx_378 : i32 loc(#loc214)
+      tt.reduce.return %right_idx_379 : i32 loc(#loc209)
+    }) : (tensor<256x2x1xi32>) -> tensor<256x1xi32> loc(#loc209)
+    %right_idx_363 = tt.expand_dims %right_idx_362 {axis = 1 : i32} : tensor<256x1xi32> -> tensor<256x1x1xi32> loc(#loc183)
+    %right_idx_364 = tt.broadcast %right_idx_363 : tensor<256x1x1xi32> -> tensor<256x2x1xi32> loc(#loc184)
+    %left_idx_365 = tt.reshape %left_idx_360 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc185)
+    %right_idx_366 = tt.reshape %right_idx_364 : tensor<256x2x1xi32> -> tensor<32x16xi32> loc(#loc186)
+    %cond_367 = arith.cmpi slt, %ileft_354, %iright_355 : tensor<32x16xi32> loc(#loc187)
+    %eq_368 = arith.cmpi eq, %ileft_354, %iright_355 : tensor<32x16xi32> loc(#loc188)
+    %cond_369 = arith.cmpi sgt, %left_idx_365, %right_idx_366 : tensor<32x16xi32> loc(#loc189)
+    %cond_370 = arith.andi %eq_368, %cond_369 : tensor<32x16xi1> loc(#loc190)
+    %cond_371 = arith.ori %cond_367, %cond_370 : tensor<32x16xi1> loc(#loc191)
+    %new_idxs_372 = arith.xori %left_idx_365, %right_idx_366 : tensor<32x16xi32> loc(#loc197)
+    %new_idxs_373 = arith.select %cond_371, %new_idxs_372, %cst_0 : tensor<32x16xi1>, tensor<32x16xi32> loc(#loc198)
+    %new_idxs_374 = arith.xori %new_idxs_344, %new_idxs_373 : tensor<32x16xi32> loc(#loc199)
+    %tmp7 = arith.extsi %tmp0_19 : tensor<32x16xi32> to tensor<32x16xi64> loc(#loc149)
+    %tmp10_375 = arith.select %tmp0_18, %tmp7, %tmp10 : tensor<32x16xi1>, tensor<32x16xi64> loc(#loc86)
+    %tmp11 = "tt.reduce"(%tmp10_375) <{axis = 1 : i32}> ({
+    ^bb0(%tmp11_377: i64 loc(callsite(#loc1 at #loc150)), %tmp11_378: i64 loc(callsite(#loc1 at #loc150))):
+      %tmp11_379 = arith.addi %tmp11_377, %tmp11_378 : i64 loc(#loc200)
+      tt.reduce.return %tmp11_379 : i64 loc(#loc160)
+    }) : (tensor<32x16xi64>) -> tensor<32xi64> loc(#loc160)
+    %tmp11_376 = tt.expand_dims %tmp11 {axis = 1 : i32} : tensor<32xi64> -> tensor<32x1xi64> loc(#loc151)
+    %tmp14 = arith.trunci %tmp11_376 : tensor<32x1xi64> to tensor<32x1xi32> loc(#loc152)
+    %0 = arith.muli %xindex_6, %cst_2 : tensor<32x1xi32> loc(#loc73)
+    %1 = tt.broadcast %r0_index_8 : tensor<1x16xi32> -> tensor<32x16xi32> loc(#loc74)
+    %2 = tt.broadcast %0 : tensor<32x1xi32> -> tensor<32x16xi32> loc(#loc74)
+    %3 = arith.addi %1, %2 : tensor<32x16xi32> loc(#loc74)
+    %4 = tt.splat %out_ptr2 : !tt.ptr<i32> -> tensor<32x16x!tt.ptr<i32>> loc(#loc75)
+    %5 = tt.addptr %4, %3 : tensor<32x16x!tt.ptr<i32>>, tensor<32x16xi32> loc(#loc75)
+    tt.store %5, %new_idxs_374, %tmp0_18 : tensor<32x16x!tt.ptr<i32>> loc(#loc76)
+    %6 = tt.splat %out_ptr3 : !tt.ptr<i32> -> tensor<32x1x!tt.ptr<i32>> loc(#loc77)
+    %7 = tt.addptr %6, %xindex_6 : tensor<32x1x!tt.ptr<i32>>, tensor<32x1xi32> loc(#loc77)
+    tt.store %7, %tmp14, %xmask_7 : tensor<32x1x!tt.ptr<i32>> loc(#loc78)
+    tt.return loc(#loc79)
+  } loc(#loc)
+} loc(#loc)
+#loc3 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":44:34)
+#loc4 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:49)
+#loc5 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:38)
+#loc6 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":26:21)
+#loc7 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":24:28)
+#loc8 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":24:33)
+#loc9 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":25:36)
+#loc10 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":25:44)
+#loc11 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":25:23)
+#loc12 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":27:28)
+#loc13 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":27:38)
+#loc14 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":33:19)
+#loc15 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":34:19)
+#loc16 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:35)
+#loc17 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:45)
+#loc18 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:30)
+#loc19 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":36:54)
+#loc20 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":38:19)
+#loc21 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":40:33)
+#loc22 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:41)
+#loc24 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:44)
+#loc25 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:60)
+#loc26 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":627:68)
+#loc27 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":533:22)
+#loc29 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":537:21)
+#loc30 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:40)
+#loc31 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":291:36)
+#loc33 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":261:15)
+#loc34 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:65)
+#loc35 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":538:78)
+#loc36 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:41)
+#loc38 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:67)
+#loc39 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":539:80)
+#loc40 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":540:30)
+#loc41 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":541:32)
+#loc42 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":546:29)
+#loc43 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:36)
+#loc44 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:23)
+#loc45 = loc("/workspace/specforge/lib/python3.11/site-packages/triton/language/standard.py":290:25)
+#loc47 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:53)
+#loc48 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":548:66)
+#loc49 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:37)
+#loc50 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:23)
+#loc52 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:54)
+#loc53 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":551:67)
+#loc54 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":553:36)
+#loc55 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":554:38)
+#loc56 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":574:22)
+#loc57 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":591:21)
+#loc58 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:40)
+#loc59 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:29)
+#loc60 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":594:23)
+#loc61 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":599:19)
+#loc62 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":599:28)
+#loc63 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:38)
+#loc64 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:46)
+#loc65 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":600:15)
+#loc66 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:48)
+#loc67 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:59)
+#loc68 = loc("/workspace/specforge/lib/python3.11/site-packages/torch/_inductor/runtime/triton_helpers.py":601:22)
+#loc69 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":42:19)
+#loc71 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":45:29)
+#loc72 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":48:21)
+#loc73 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:35)
+#loc74 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:32)
+#loc75 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:25)
+#loc76 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":49:47)
+#loc77 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:25)
+#loc78 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:37)
+#loc79 = loc("/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/ce/ccefvxdobtenaocabewevaf45h6p7aehp54xd7ccf3732l3vxvwg.py":50:4)
+#loc85 = loc(callsite(#loc1 at #loc2))
+#loc86 = loc("tmp10"(#loc3))
+#loc87 = loc("tmp0"(#loc4))
+#loc88 = loc("tmp0"(#loc5))
+#loc89 = loc("xmask"(#loc6))
+#loc90 = loc("xoffset"(#loc7))
+#loc91 = loc("xoffset"(#loc8))
+#loc92 = loc("xindex"(#loc9))
+#loc93 = loc("xindex"(#loc10))
+#loc94 = loc("xindex"(#loc11))
+#loc95 = loc("r0_index"(#loc12))
+#loc96 = loc("r0_index"(#loc13))
+#loc97 = loc("x0"(#loc14))
+#loc98 = loc("x1"(#loc15))
+#loc99 = loc("tmp0"(#loc16))
+#loc100 = loc("tmp0"(#loc17))
+#loc101 = loc("tmp0"(#loc18))
+#loc102 = loc("tmp0"(#loc19))
+#loc103 = loc("tmp2"(#loc20))
+#loc104 = loc("tmp4"(#loc21))
+#loc105 = loc("flip"(#loc22))
+#loc107 = loc("flip"(#loc24))
+#loc108 = loc("flip"(#loc25))
+#loc109 = loc("flip"(#loc26))
+#loc110 = loc("y"(#loc27))
+#loc111 = loc("left_mask"(#loc29))
+#loc112 = loc("ileft"(#loc30))
+#loc114 = loc("ileft"(#loc34))
+#loc115 = loc("ileft"(#loc35))
+#loc116 = loc("iright"(#loc36))
+#loc118 = loc("iright"(#loc38))
+#loc119 = loc("iright"(#loc39))
+#loc120 = loc("ileft"(#loc40))
+#loc121 = loc("iright"(#loc41))
+#loc122 = loc("y_idx"(#loc42))
+#loc123 = loc("left_idx"(#loc43))
+#loc124 = loc("left_idx"(#loc44))
+#loc125 = loc("input"(#loc45))
+#loc127 = loc("left_idx"(#loc47))
+#loc128 = loc("left_idx"(#loc48))
+#loc129 = loc("right_idx"(#loc49))
+#loc130 = loc("right_idx"(#loc50))
+#loc132 = loc("right_idx"(#loc52))
+#loc133 = loc("right_idx"(#loc53))
+#loc134 = loc("left_idx"(#loc54))
+#loc135 = loc("right_idx"(#loc55))
+#loc136 = loc("cond"(#loc56))
+#loc137 = loc("eq"(#loc57))
+#loc138 = loc("cond"(#loc58))
+#loc139 = loc("cond"(#loc59))
+#loc140 = loc("cond"(#loc60))
+#loc141 = loc("cond"(#loc61))
+#loc142 = loc("cond"(#loc62))
+#loc143 = loc("ret"(#loc63))
+#loc144 = loc("ret"(#loc64))
+#loc145 = loc("ret"(#loc65))
+#loc146 = loc("new_idxs"(#loc66))
+#loc147 = loc("new_idxs"(#loc67))
+#loc148 = loc("new_idxs"(#loc68))
+#loc149 = loc("tmp7"(#loc69))
+#loc151 = loc("tmp11"(#loc71))
+#loc152 = loc("tmp14"(#loc72))
+#loc153 = loc(callsite(#loc105 at #loc106))
+#loc154 = loc(callsite(#loc107 at #loc106))
+#loc155 = loc(callsite(#loc108 at #loc106))
+#loc156 = loc(callsite(#loc109 at #loc106))
+#loc158 = loc("cond"(#loc136))
+#loc159 = loc("eq"(#loc137))
+#loc160 = loc(callsite(#loc31 at #loc150))
+#loc162 = loc(callsite(#loc110 at #loc157))
+#loc163 = loc(callsite(#loc111 at #loc157))
+#loc164 = loc(callsite(#loc112 at #loc157))
+#loc166 = loc(callsite(#loc114 at #loc157))
+#loc167 = loc(callsite(#loc115 at #loc157))
+#loc168 = loc(callsite(#loc116 at #loc157))
+#loc170 = loc(callsite(#loc118 at #loc157))
+#loc171 = loc(callsite(#loc119 at #loc157))
+#loc172 = loc(callsite(#loc120 at #loc157))
+#loc173 = loc(callsite(#loc121 at #loc157))
+#loc174 = loc(callsite(#loc122 at #loc157))
+#loc175 = loc(callsite(#loc123 at #loc157))
+#loc176 = loc(callsite(#loc124 at #loc157))
+#loc178 = loc(callsite(#loc127 at #loc157))
+#loc179 = loc(callsite(#loc128 at #loc157))
+#loc180 = loc(callsite(#loc129 at #loc157))
+#loc181 = loc(callsite(#loc130 at #loc157))
+#loc183 = loc(callsite(#loc132 at #loc157))
+#loc184 = loc(callsite(#loc133 at #loc157))
+#loc185 = loc(callsite(#loc134 at #loc157))
+#loc186 = loc(callsite(#loc135 at #loc157))
+#loc187 = loc(callsite(#loc158 at #loc157))
+#loc188 = loc(callsite(#loc159 at #loc157))
+#loc189 = loc(callsite(#loc138 at #loc157))
+#loc190 = loc(callsite(#loc139 at #loc157))
+#loc191 = loc(callsite(#loc140 at #loc157))
+#loc192 = loc(callsite(#loc141 at #loc157))
+#loc193 = loc(callsite(#loc142 at #loc157))
+#loc194 = loc(callsite(#loc143 at #loc157))
+#loc195 = loc(callsite(#loc144 at #loc157))
+#loc196 = loc(callsite(#loc145 at #loc157))
+#loc197 = loc(callsite(#loc146 at #loc157))
+#loc198 = loc(callsite(#loc147 at #loc157))
+#loc199 = loc(callsite(#loc148 at #loc157))
+#loc200 = loc(callsite(#loc33 at #loc160))
+#loc201 = loc(callsite(#loc31 at #loc165))
+#loc203 = loc(callsite(#loc31 at #loc169))
+#loc205 = loc(callsite(#loc125 at #loc177))
+#loc206 = loc(callsite(#loc31 at #loc177))
+#loc208 = loc(callsite(#loc125 at #loc182))
+#loc209 = loc(callsite(#loc31 at #loc182))
+#loc211 = loc(callsite(#loc33 at #loc201))
+#loc212 = loc(callsite(#loc33 at #loc203))
+#loc213 = loc(callsite(#loc33 at #loc206))
+#loc214 = loc(callsite(#loc33 at #loc209))

SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/__grp__triton_poi_fused_new_zeros_1.json

@@ -0,0 +1 @@
+{"child_paths": {"triton_poi_fused_new_zeros_1.source": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.source", "triton_poi_fused_new_zeros_1.ttir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.ttir", "triton_poi_fused_new_zeros_1.ttgir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.ttgir", "triton_poi_fused_new_zeros_1.llir": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.llir", "triton_poi_fused_new_zeros_1.ptx": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.ptx", "triton_poi_fused_new_zeros_1.cubin": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.cubin", "triton_poi_fused_new_zeros_1.json": "/workspace/hanrui/SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.json"}}

SpecForge-ext/cache/compiled_kernels/triton/0/4UWYNBR3KPWQGNAZ5LIIRE7YAZWTQP4CP3JS6GOSLWYDF5K7WTAA/triton_poi_fused_new_zeros_1.json

@@ -0,0 +1 @@
+{"hash": "e52d86863b53ed033419ead08893f8066d383f827ed32f19d25db032f55fb4c0", "target": {"backend": "cuda", "arch": 90, "warp_size": 32}, "num_warps": 4, "num_ctas": 1, "num_stages": 1, "warp_size": 32, "maxnreg": null, "cluster_dims": [1, 1, 1], "ptx_version": null, "ptx_options": null, "ir_override": null, "enable_fp_fusion": true, "launch_cooperative_grid": false, "launch_pdl": false, "supported_fp8_dtypes": ["fp8e4b15", "fp8e4nv", "fp8e5"], "deprecated_fp8_dot_operand_dtypes": ["fp8e4b15"], "default_dot_input_precision": "tf32", "allowed_dot_input_precisions": ["tf32", "tf32x3", "ieee"], "max_num_imprecise_acc_default": 1073741824, "extern_libs": [["libdevice", "/workspace/specforge/lib/python3.11/site-packages/triton/backends/nvidia/lib/libdevice.10.bc"]], "debug": true, "backend_name": "cuda", "sanitize_overflow": false, "arch": "sm90", "instrumentation_mode": "", "triton_version": "3.5.1", "tensordesc_meta": [], "shared": 0, "tmem_size": 0, "global_scratch_size": 0, "global_scratch_align": 1, "profile_scratch_size": 0, "profile_scratch_align": 1, "name": "triton_poi_fused_new_zeros_1"}