trunk/29343e76e2342c777bb257a270bae1d805996700: Support non-builtin constexpr objects in user Triton kernels (#189692)
- PyTorch 近 90 天出现 180 次
- PyTorch 近 90 天第 178 次发版
- 上一次：同一天稍早 · trunk/4687d3465afbe4b55b85296aecdec528d710c3af: [dynamo] Keep the subclass type when resyncing tensor metadata (#193969)
发生了什么
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…
摘要按规则整理自下方来源原文