feat: triton lowering [take III]

[avik-pal]

No description provided.

[avik-pal]

julia: external/llvm-project/llvm/lib/MC/MCSubtargetInfo.cpp:63: void ApplyFeatureFlag(FeatureBitset &, StringRef, ArrayRef<SubtargetFeatureKV>): Assertion `SubtargetFeatures::hasFlag(Feature) && "Feature flags should start with '+' or '-'"' failed.

[659431] signal 6 (-6): Aborted
in expression starting at /mnt2/avik-pal/reactant/Reactant.jl/envs/jax_integrate/triton.jl:88
pthread_kill at /lib/x86_64-linux-gnu/libc.so.6 (unknown line)
gsignal at /lib/x86_64-linux-gnu/libc.so.6 (unknown line)
abort at /lib/x86_64-linux-gnu/libc.so.6 (unknown line)
unknown function (ip: 0x7350cd42881a)
__assert_fail at /lib/x86_64-linux-gnu/libc.so.6 (unknown line)
ApplyFeatureFlag at /proc/self/cwd/external/llvm-project/llvm/lib/MC/MCSubtargetInfo.cpp:62
getFeatures at /proc/self/cwd/external/llvm-project/llvm/lib/MC/MCSubtargetInfo.cpp:224
InitMCProcessorInfo at /proc/self/cwd/external/llvm-project/llvm/lib/MC/MCSubtargetInfo.cpp:233
MCSubtargetInfo at /proc/self/cwd/external/llvm-project/llvm/lib/MC/MCSubtargetInfo.cpp:259
TargetSubtargetInfo at /proc/self/cwd/external/llvm-project/llvm/lib/CodeGen/TargetSubtargetInfo.cpp:23
NVPTXGenSubtargetInfo at /proc/self/cwd/bazel-out/k8-dbg/bin/external/llvm-project/llvm/_virtual_includes/NVPTXCommonTableGen/NVPTXGenSubtargetInfo.inc:496
NVPTXSubtarget at /proc/self/cwd/external/llvm-project/llvm/lib/Target/NVPTX/NVPTXSubtarget.cpp:63
NVPTXTargetMachine at /proc/self/cwd/external/llvm-project/llvm/lib/Target/NVPTX/NVPTXTargetMachine.cpp:134
NVPTXTargetMachine64 at /proc/self/cwd/external/llvm-project/llvm/lib/Target/NVPTX/NVPTXTargetMachine.cpp:165
Allocator at /proc/self/cwd/external/llvm-project/llvm/include/llvm/MC/TargetRegistry.h:1314
createTargetMachine at /proc/self/cwd/external/llvm-project/llvm/include/llvm/MC/TargetRegistry.h:498
getOrCreateTargetMachine at /proc/self/cwd/external/llvm-project/mlir/lib/Target/LLVM/ModuleToObject.cpp:70
setDataLayoutAndTriple at /proc/self/cwd/external/llvm-project/mlir/lib/Target/LLVM/ModuleToObject.cpp:230
run at /proc/self/cwd/external/llvm-project/mlir/lib/Target/LLVM/ModuleToObject.cpp:255
serializeToObject at /proc/self/cwd/external/llvm-project/mlir/lib/Target/LLVM/NVVM/Target.cpp:750
serializeToObject at /proc/self/cwd/bazel-out/k8-dbg/bin/external/llvm-project/mlir/include/mlir/Dialect/GPU/IR/CompilationAttrInterfaces.h.inc:249
serializeToObject at /proc/self/cwd/bazel-out/k8-dbg/bin/external/llvm-project/mlir/include/mlir/Dialect/GPU/IR/CompilationAttrInterfaces.cpp.inc:19
moduleSerializer at /proc/self/cwd/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ModuleToBinary.cpp:91
transformGpuModulesToBinaries at /proc/self/cwd/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ModuleToBinary.cpp:125
runOnOperation at /proc/self/cwd/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ModuleToBinary.cpp:69
operator() at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:609
callback_fn<(lambda at external/llvm-project/mlir/lib/Pass/Pass.cpp:604:7)> at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:46
operator() at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:69
executeAction<mlir::PassExecutionAction, mlir::Pass &> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/MLIRContext.h:290
run at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:603
runPipeline at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:682
runPasses at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:1117
run at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:1091
CompileCall at /proc/self/cwd/external/enzyme_ad/src/enzyme_ad/jax/Passes/LowerJIT.cpp:891
operator() at /proc/self/cwd/external/enzyme_ad/src/enzyme_ad/jax/Passes/LowerJIT.cpp:1006
operator() at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Visitors.h:304
callback_fn<(lambda at external/llvm-project/mlir/include/mlir/IR/Visitors.h:302:20)> at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:46
operator() at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:69
walk<mlir::ForwardIterator> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Visitors.h:152
walk<mlir::ForwardIterator> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Visitors.h:147
walk<mlir::ForwardIterator> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Visitors.h:147
walk<(mlir::WalkOrder)1, mlir::ForwardIterator, (lambda at external/enzyme_ad/src/enzyme_ad/jax/Passes/LowerJIT.cpp:978:26), mlir::enzymexla::JITCallOp, void> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Visitors.h:306
walk<(mlir::WalkOrder)1, mlir::ForwardIterator, (lambda at external/enzyme_ad/src/enzyme_ad/jax/Passes/LowerJIT.cpp:978:26), void> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/Operation.h:798
runOnOperation at /proc/self/cwd/external/enzyme_ad/src/enzyme_ad/jax/Passes/LowerJIT.cpp:978
operator() at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:609
callback_fn<(lambda at external/llvm-project/mlir/lib/Pass/Pass.cpp:604:7)> at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:46
operator() at /proc/self/cwd/external/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:69
executeAction<mlir::PassExecutionAction, mlir::Pass &> at /proc/self/cwd/external/llvm-project/mlir/include/mlir/IR/MLIRContext.h:290
run at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:603
runPipeline at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:682
runPasses at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:1117
run at /proc/self/cwd/external/llvm-project/mlir/lib/Pass/Pass.cpp:1091
mlirPassManagerRunOnOp at /proc/self/cwd/external/llvm-project/mlir/lib/CAPI/IR/Pass.cpp:45
mlirPassManagerRunOnOp at /mnt2/avik-pal/reactant/Reactant.jl/src/mlir/libMLIR_h.jl:8951 [inlined]
run! at /mnt2/avik-pal/reactant/Reactant.jl/src/mlir/IR/Pass.jl:154
#run_pass_pipeline!#2 at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:1319
run_pass_pipeline! at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:1314 [inlined]
#compile_mlir!#15 at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:1770
compile_mlir! at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:1576 [inlined]
compile_mlir! at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:1576 [inlined]
#compile_xla#58 at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:3582
compile_xla at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:3554 [inlined]
#compile#59 at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:3658
compile at /mnt2/avik-pal/reactant/Reactant.jl/src/Compiler.jl:3655
unknown function (ip: 0x734dfe1db1ad)
jl_apply at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/julia.h:2157 [inlined]
do_call at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:126
eval_value at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:223
eval_stmt_value at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:174 [inlined]
eval_body at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:666
jl_interpret_toplevel_thunk at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:824
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:943
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:886
ijl_toplevel_eval_in at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:994
eval at ./boot.jl:430 [inlined]
include_string at ./loading.jl:2734
_include at ./loading.jl:2794
include at ./sysimg.jl:38
unknown function (ip: 0x734ebf323fe2)
jl_apply at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/julia.h:2157 [inlined]
do_call at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:126
eval_value at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:223
eval_stmt_value at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:174 [inlined]
eval_body at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:666
jl_interpret_toplevel_thunk at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/interpreter.c:824
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:943
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:886
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:886
jl_toplevel_eval_flex at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:886
ijl_toplevel_eval_in at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/toplevel.c:994
eval at ./boot.jl:430 [inlined]
eval_user_input at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:261
repl_backend_loop at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:368
#start_repl_backend#59 at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:343
start_repl_backend at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:340
#run_repl#76 at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:500
run_repl at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/usr/share/julia/stdlib/v1.11/REPL/src/REPL.jl:486
jfptr_run_repl_10226.1 at /mnt2/avik-pal/.julia/juliaup/julia-1.11.7+0.x64.linux.gnu/share/julia/compiled/v1.11/REPL/u0gqU_XvZAg.so (unknown line)
#1150 at ./client.jl:446
jfptr_YY.1150_14873.1 at /mnt2/avik-pal/.julia/juliaup/julia-1.11.7+0.x64.linux.gnu/share/julia/compiled/v1.11/REPL/u0gqU_XvZAg.so (unknown line)
jl_apply at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/julia.h:2157 [inlined]
jl_f__call_latest at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/builtins.c:875
#invokelatest#2 at ./essentials.jl:1055 [inlined]
invokelatest at ./essentials.jl:1052 [inlined]
run_main_repl at ./client.jl:430
repl_main at ./client.jl:567 [inlined]
_start at ./client.jl:541
jfptr__start_73684.1 at /mnt2/avik-pal/.julia/juliaup/julia-1.11.7+0.x64.linux.gnu/lib/julia/sys.so (unknown line)
jl_apply at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/julia.h:2157 [inlined]
true_main at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/jlapi.c:900
jl_repl_entrypoint at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/src/jlapi.c:1059
main at /cache/build/builder-demeter6-3/julialang/julia-release-1-dot-11/cli/loader_exe.c:58
unknown function (ip: 0x7350cd42a1c9)
__libc_start_main at /lib/x86_64-linux-gnu/libc.so.6 (unknown line)
unknown function (ip: 0x4010b8)
Allocations: 92540992 (Pool: 92537930; Big: 3062); GC: 55
Aborted (core dumped)