CUPY_CACHE_SAVE_CUDA_SOURCE=1
CUPY_DUMP_CUDA_SOURCE_ON_ERROR=1
CUPY_ACCELERATORS='cub,cutensor'
NVCC_PATH='/usr/local/cuda/bin/nvcc'