Upload folder using huggingface_hub

modules/components/upr_net_freq/costvol.py

@@ -26,7 +26,7 @@ def cuda_float32(fltIn:float):
 
 def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
     if 'device' not in objCudacache:
-        objCudacache['device'] = "cpu"#torch.cuda.get_device_name()
+        objCudacache['device'] = "-1"#torch.cuda.get_device_name()
     # end
 
     strKey = strFunction
@@ -216,9 +216,9 @@ def cuda_kernel(strFunction:str, strKernel:str, objVariables:typing.Dict):
 # end
 
 
-@cupy.memoize(for_each_device=True)
+@cupy.memoize(for_each_device=False)#True)
 def cuda_launch(strKey:str):
-    os.environ["CUDA_VISIBLE_DEVICES"]=""
+
     if 'CUDA_HOME' not in os.environ:
         os.environ['CUDA_HOME'] = '/usr/local/cuda/'
     # end