[asukaminato0721]

Summary

part of #3884, Pyrefly now treats tl.constexpr parameters as Any in recognized Triton JIT functions.

• I am a human, and I fully understand the code I am submitting.

Test Plan

add test

[meta-codesync]

This pull request has been imported. If you are a Meta employee, you can view this in D126647218. (Because this pull request was imported automatically, there will not be any future comments.)

[stroxler]

I'll take a look at this, thanks!

I suspect we'll wind up treating this more as a prototype than something to merge quickly, because triton support is a big undertaking and we probably want to tie it to tensor shape types (which also will mean designing an extensive stub library).

[github-actions]

According to mypy_primer, this change doesn't affect type check results on a corpus of open source code. ✅

[rchen152]

Sounds like this is more of a prototype/waiting for interactions with tensor shapes to be worked out, so I'm going to move it to draft so I don't keep seeing it and thinking it needs review :)