trunk/29343e76e2342c777bb257a270bae1d805996700: Support non-builtin constexpr objects in user Triton kernels (#189692)
- PyTorch: 180 events in the last 90 days
- PyTorch: 178th Release in the last 90 days
- Previous: earlier the same day · trunk/4687d3465afbe4b55b85296aecdec528d710c3af: [dynamo] Keep the subclass type when resyncing tensor metadata (#193969)
What happened
Fix Inductor codegen for user-defined Triton kernels that pass non-builtin Python objects as tl.constexpr values. When such constants are serialized with constructor-style repr output, for example UserDefinedTritonKernelConfig(offset=2) , generated kernel source and launcher code need the corresponding class in scope. This change imports the constant class into generated kernel source and makes it available in the g…
Summary assembled by rule from the sources below