Use PyLong_AsVoidPtr to cast dev_ptr (#2192)

Fixes: #2188 Calling PyLong_AsLongLong causes overflow and return of a wrong pointer. Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>
intel · Sep 11, 2024 · b6e5fd2 · b6e5fd2
1 parent 3102cf5
commit b6e5fd2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/benchmarks/triton_kernels_benchmark/benchmark_driver.py b/benchmarks/triton_kernels_benchmark/benchmark_driver.py
@@ -204,7 +204,7 @@ def format_of(ty):
       ptr_info.dev_ptr = 0;
       ptr_info.valid = true;
       if (PyLong_Check(obj)) {{
-        ptr_info.dev_ptr = (void*) PyLong_AsLongLong(obj);
+        ptr_info.dev_ptr = PyLong_AsVoidPtr(obj)
         checkDevicePointer(&ptr_info, idx, queue);
         return ptr_info;
       }}
@@ -223,7 +223,7 @@ def format_of(ty):
           ptr_info.valid = false;
           return ptr_info;
         }}
-        ptr_info.dev_ptr = (void*) PyLong_AsLongLong(ret);
+        ptr_info.dev_ptr = PyLong_AsVoidPtr(ret);
         if(!ptr_info.dev_ptr) {{
           return ptr_info;
         }}

diff --git a/third_party/intel/backend/driver.py b/third_party/intel/backend/driver.py
@@ -271,7 +271,7 @@ def format_of(ty):
       ptr_info.dev_ptr = 0;
       ptr_info.valid = true;
       if (PyLong_Check(obj)) {{
-        ptr_info.dev_ptr = (void*) PyLong_AsLongLong(obj);
+        ptr_info.dev_ptr = PyLong_AsVoidPtr(obj);
         checkDevicePointer(&ptr_info, idx, queue);
         return ptr_info;
       }}
@@ -290,7 +290,7 @@ def format_of(ty):
           ptr_info.valid = false;
           return ptr_info;
         }}
-        ptr_info.dev_ptr = (void*) PyLong_AsLongLong(ret);
+        ptr_info.dev_ptr = PyLong_AsVoidPtr(ret);
         if(!ptr_info.dev_ptr) {{
           return ptr_info;
         }}