From d2544e7c559b9dcbda326820921c6263a3ecff2c Mon Sep 17 00:00:00 2001 From: Anthony Berg Date: Tue, 24 Jun 2025 17:45:50 +0200 Subject: [PATCH] fix(kernel): type for defines in get_module --- GPUSimulators/gpu/cuda_context.py | 2 +- GPUSimulators/gpu/hip_context.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/GPUSimulators/gpu/cuda_context.py b/GPUSimulators/gpu/cuda_context.py index a9d6219..fc09bac 100644 --- a/GPUSimulators/gpu/cuda_context.py +++ b/GPUSimulators/gpu/cuda_context.py @@ -182,7 +182,7 @@ class CudaContext(object): def get_module(self, kernel_filename: str, include_dirs: dict=None, - defines:list[str]=None, + defines:dict[str: int]=None, compile_args:dict=None, jit_compile_args:dict=None) -> cuda.Module: """ Reads a text file and creates an OpenCL kernel from that. diff --git a/GPUSimulators/gpu/hip_context.py b/GPUSimulators/gpu/hip_context.py index 72d1647..39e9838 100644 --- a/GPUSimulators/gpu/hip_context.py +++ b/GPUSimulators/gpu/hip_context.py @@ -58,7 +58,7 @@ class HIPContext(Context): def get_module(self, kernel_filename: str, include_dirs: dict=None, - defines:list[str]=None, + defines:dict[str: int]=None, compile_args:dict=None, jit_compile_args:dict=None): """