Skip to content

Commit 7bdc903

Browse files
committed
refactor: remove unused device property from Function
1 parent 2309b7d commit 7bdc903

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

numba_cuda/numba/cuda/cudadrv/driver.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2318,7 +2318,7 @@ def get_function(self, name):
23182318
driver.cuModuleGetFunction(
23192319
byref(handle), self.handle, name.encode("utf8")
23202320
)
2321-
return CtypesFunction(weakref.proxy(self), handle, name)
2321+
return CtypesFunction(handle, name)
23222322

23232323
def get_global_symbol(self, name):
23242324
ptr = drvapi.cu_device_ptr()
@@ -2332,7 +2332,7 @@ def get_global_symbol(self, name):
23322332
class CudaPythonModule(Module):
23332333
def get_function(self, name):
23342334
handle = driver.cuModuleGetFunction(self.handle, name.encode("utf8"))
2335-
return CudaPythonFunction(weakref.proxy(self), handle, name)
2335+
return CudaPythonFunction(handle, name)
23362336

23372337
def get_global_symbol(self, name):
23382338
ptr, size = driver.cuModuleGetGlobal(self.handle, name.encode("utf8"))
@@ -2350,19 +2350,14 @@ class Function(metaclass=ABCMeta):
23502350
stream = 0
23512351
sharedmem = 0
23522352

2353-
def __init__(self, module, handle, name):
2354-
self.module = module
2353+
def __init__(self, handle, name):
23552354
self.handle = handle
23562355
self.name = name
23572356
self.attrs = self.read_func_attr_all()
23582357

23592358
def __repr__(self):
23602359
return "<CUDA function %s>" % self.name
23612360

2362-
@property
2363-
def device(self):
2364-
return self.module.context.device
2365-
23662361
@abstractmethod
23672362
def cache_config(
23682363
self, prefer_equal=False, prefer_cache=False, prefer_shared=False

0 commit comments

Comments
 (0)