mirror of
https://github.com/smyalygames/FiniteVolumeGPU.git
synced 2025-09-14 11:12:17 +02:00
fix(kernel): type for defines in get_module
This commit is contained in:
parent
ae23145753
commit
d2544e7c55
@ -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.
|
||||
|
@ -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):
|
||||
"""
|
||||
|
Loading…
x
Reference in New Issue
Block a user