@@ -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):
23322332class 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