diff --git a/ds4_cuda.cu b/ds4_cuda.cu index 8b6241ca3..5db8b15b2 100644 --- a/ds4_cuda.cu +++ b/ds4_cuda.cu @@ -1430,6 +1430,11 @@ extern "C" int ds4_gpu_set_model_map(const void *model_map, uint64_t model_size) } } + const char *chunked_env = getenv("DS4_CUDA_COPY_MODEL_CHUNKED"); + if (chunked_env && chunked_env[0]) { + return 1; + } + cudaError_t err = cudaHostRegister((void *)model_map, (size_t)model_size, cudaHostRegisterMapped | cudaHostRegisterReadOnly); if (err == cudaSuccess) {