aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJoshua Bakita <bakitajoshua@gmail.com>2025-05-20 11:01:09 -0400
committerJoshua Bakita <bakitajoshua@gmail.com>2025-05-20 11:01:09 -0400
commit03ae77e35d35b2a82f5387d1903cfa954b696edd (patch)
tree793fcdbac376abd987507ae5210c242221c246f7
parent97e97dc9d55e71edbc2031e6a509a7cc17abe168 (diff)
Correct how path of libsmctrl.so is specified in LD_PRELOAD by nvtaskset
Indicate that the loader should search the standard library paths by ommiting the "./" prefix.
-rw-r--r--nvtaskset.c2
1 files changed, 1 insertions, 1 deletions
diff --git a/nvtaskset.c b/nvtaskset.c
index b8873ab..74f88d7 100644
--- a/nvtaskset.c
+++ b/nvtaskset.c
@@ -134,7 +134,7 @@ int main(int argc, char **argv) {
134 } 134 }
135 // Tell loader to initialize libsmctrl.so first 135 // Tell loader to initialize libsmctrl.so first
136 // TODO: Append, rather than overwrite LD_PRELOAD 136 // TODO: Append, rather than overwrite LD_PRELOAD
137 setenv("LD_PRELOAD", "./libsmctrl.so", 1); 137 setenv("LD_PRELOAD", "libsmctrl.so", 1);
138 // Explictly set the number of channels, otherwise CUDA will only use two 138 // Explictly set the number of channels, otherwise CUDA will only use two
139 // (see paper for why that causes problems) 139 // (see paper for why that causes problems)
140 setenv("CUDA_DEVICE_MAX_CONNECTIONS", "8", 1); 140 setenv("CUDA_DEVICE_MAX_CONNECTIONS", "8", 1);