From 160a50af77305964c75fe7137d640435f85bd7f1 Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Tue, 18 Jul 2023 16:18:56 +0200 Subject: [PATCH] cleanup --- launcher/src/main.rs | 6 ------ 1 file changed, 6 deletions(-) diff --git a/launcher/src/main.rs b/launcher/src/main.rs index f4c11211..c03af4b7 100644 --- a/launcher/src/main.rs +++ b/launcher/src/main.rs @@ -369,12 +369,6 @@ fn shard_manager( // Copy current process env let mut envs: Vec<(OsString, OsString)> = env::vars_os().collect(); - // Use cuda allocator. It leads to less memory fragmentation - // envs.push(( - // "PYTORCH_CUDA_ALLOC_CONF".into(), - // "backend:cudaMallocAsync".into(), - // )); - // Torch Distributed Env vars envs.push(("RANK".into(), rank.to_string().into())); envs.push(("WORLD_SIZE".into(), world_size.to_string().into()));