cupy.cuda.compile_with_cache deprecated

[ricshaw]

Hi, thanks for the great work. I am trying to run the demo code.
The function cupy.cuda.compile_with_cache() is deprecated in cupy. How do we update this?

[m43]

Maybe you can try using the cupy version reported in the requirements.txt, worked fine for me:

SpaTracker/requirements.txt

Line 10 in 4e777d1

cupy==12.2.0

[thanhdat77]

Hi @ricshaw,

I tried cupy-cuda11x==12.2.0, and it worked for me. It seems that the auto setup in CuPy may incorrectly detect your CUDA version.

If you have CUDA Toolkit 12 installed, try using:

cupy-cuda12x==12.2.0

[junbangliang]

I replaced the function that caused the error and ran successfully on cupy==13.3.0 with cuda 12.5:

@cupy.memoize(for_each_device=True)
def cuda_launch(strKey:str):
    if 'CUDA_HOME' not in os.environ:
        os.environ['CUDA_HOME'] = cupy.cuda.get_cuda_path()
    # end

    return cupy.cuda.compile_with_cache(objCudacache[strKey]['strKernel'], tuple(['-I ' + os.environ['CUDA_HOME'], '-I ' + os.environ['CUDA_HOME'] + '/include'])).get_function(objCudacache[strKey]['strFunction'])
# end

replaced with:

@cupy.memoize(for_each_device=True)
def cuda_launch(strKey: str):
    if 'CUDA_HOME' not in os.environ:
        os.environ['CUDA_HOME'] = cupy.cuda.get_cuda_path()

    # Compile the kernel using cupy.RawModule instead of compile_with_cache
    kernel_code = objCudacache[strKey]['strKernel']
    module = cupy.RawModule(code=kernel_code, options=('-I', os.environ['CUDA_HOME'], '-I', os.environ['CUDA_HOME'] + '/include'))
    
    # Get the kernel function from the compiled module
    kernel_function = module.get_function(objCudacache[strKey]['strFunction'])

    return kernel_function
# end