From 245b1e4484aa944013d9897e2472568bfa5cca88 Mon Sep 17 00:00:00 2001 From: Jeremy Lorelli Date: Tue, 20 Feb 2024 17:12:08 -0800 Subject: [PATCH] Set NVreg_EnableStreamMemOPs=1 when loading nvidia.ko Needed to enable GPUDirect feature set in usermode --- data_gpu/driver/comp_and_load_drivers.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_gpu/driver/comp_and_load_drivers.sh b/data_gpu/driver/comp_and_load_drivers.sh index 798409f..3b552ff 100755 --- a/data_gpu/driver/comp_and_load_drivers.sh +++ b/data_gpu/driver/comp_and_load_drivers.sh @@ -13,7 +13,7 @@ cd $NVIDIA_PATH make CC=gcc-12 -/usr/sbin/insmod nvidia.ko NVreg_OpenRmEnableUnsupportedGpus=1 +/usr/sbin/insmod nvidia.ko NVreg_OpenRmEnableUnsupportedGpus=1 NVreg_EnableStreamMemOPs=1 /usr/sbin/insmod nvidia-modeset.ko /usr/sbin/insmod nvidia-uvm.ko /usr/sbin/insmod nvidia-drm.ko modeset=1