[easy] Fix unsigned long issue in static cuda launcher (#162920)

Fixes https://github.com/pytorch/pytorch/issues/162430

It's a bit hard to come up with a unit test where the stream exceeds a C++ long, so just using existing unit tests for now.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162920
Approved by: https://github.com/Skylion007, https://github.com/jansel
This commit is contained in:
James Wu
2025-09-14 12:47:38 -07:00
committed by PyTorch MergeBot
parent 09cbf34e93
commit 7d1bcd9aea

View File

@ -369,7 +369,7 @@ PyObject* launch_kernel(PyObject* self, PyObject* args) {
// Parse the fixed arguments and the format string
if (!PyArg_ParseTuple(
args,
"KiiiiisOl",
"KiiiiisOK",
&func_ptr,
&gridX,
&gridY,