From bef565be9438f3b56d3517c48a48426a556c9f27 Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Thu, 5 Dec 2024 14:40:28 +0100 Subject: [PATCH 1/2] Update integer literals for 'PyArg_ParseTuple' Signed-off-by: Anatoly Myachev --- benchmarks/triton_kernels_benchmark/benchmark_driver.py | 4 ++-- third_party/amd/backend/driver.py | 4 ++-- third_party/intel/backend/driver.py | 2 +- third_party/nvidia/backend/driver.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/benchmarks/triton_kernels_benchmark/benchmark_driver.py b/benchmarks/triton_kernels_benchmark/benchmark_driver.py index 2e0a484f06..a89c91c53a 100644 --- a/benchmarks/triton_kernels_benchmark/benchmark_driver.py +++ b/benchmarks/triton_kernels_benchmark/benchmark_driver.py @@ -132,8 +132,8 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "i", - "int64_t": "l", + "int32_t": "l", + "int64_t": "L", "uint8_t": "B", "uint16_t": "H", "uint32_t": "I", diff --git a/third_party/amd/backend/driver.py b/third_party/amd/backend/driver.py index b10def16bc..0b67e57772 100644 --- a/third_party/amd/backend/driver.py +++ b/third_party/amd/backend/driver.py @@ -219,8 +219,8 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "i", - "int64_t": "l", + "int32_t": "l", + "int64_t": "L", "uint8_t": "B", "uint16_t": "H", "uint32_t": "I", diff --git a/third_party/intel/backend/driver.py b/third_party/intel/backend/driver.py index 9510e73b9c..c6d54cbe9b 100644 --- a/third_party/intel/backend/driver.py +++ b/third_party/intel/backend/driver.py @@ -217,7 +217,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "i", + "int32_t": "l", "int64_t": "L", "uint8_t": "B", "uint16_t": "H", diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py index e437d53a4a..3251d7a5e8 100644 --- a/third_party/nvidia/backend/driver.py +++ b/third_party/nvidia/backend/driver.py @@ -140,7 +140,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "i", + "int32_t": "l", "int64_t": "L", "uint8_t": "B", "uint16_t": "H", From e4d3d566c096d85983efd2af711fe3757983efba Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Thu, 5 Dec 2024 18:17:22 +0100 Subject: [PATCH 2/2] revert change for int32_t Signed-off-by: Anatoly Myachev --- benchmarks/triton_kernels_benchmark/benchmark_driver.py | 2 +- third_party/amd/backend/driver.py | 2 +- third_party/intel/backend/driver.py | 2 +- third_party/nvidia/backend/driver.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/benchmarks/triton_kernels_benchmark/benchmark_driver.py b/benchmarks/triton_kernels_benchmark/benchmark_driver.py index a89c91c53a..d9b55cc988 100644 --- a/benchmarks/triton_kernels_benchmark/benchmark_driver.py +++ b/benchmarks/triton_kernels_benchmark/benchmark_driver.py @@ -132,7 +132,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "l", + "int32_t": "i", "int64_t": "L", "uint8_t": "B", "uint16_t": "H", diff --git a/third_party/amd/backend/driver.py b/third_party/amd/backend/driver.py index 0b67e57772..99e5509eca 100644 --- a/third_party/amd/backend/driver.py +++ b/third_party/amd/backend/driver.py @@ -219,7 +219,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "l", + "int32_t": "i", "int64_t": "L", "uint8_t": "B", "uint16_t": "H", diff --git a/third_party/intel/backend/driver.py b/third_party/intel/backend/driver.py index c6d54cbe9b..9510e73b9c 100644 --- a/third_party/intel/backend/driver.py +++ b/third_party/intel/backend/driver.py @@ -217,7 +217,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "l", + "int32_t": "i", "int64_t": "L", "uint8_t": "B", "uint16_t": "H", diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py index 3251d7a5e8..e437d53a4a 100644 --- a/third_party/nvidia/backend/driver.py +++ b/third_party/nvidia/backend/driver.py @@ -140,7 +140,7 @@ def format_of(ty): "long": "l", "int8_t": "b", "int16_t": "h", - "int32_t": "l", + "int32_t": "i", "int64_t": "L", "uint8_t": "B", "uint16_t": "H",