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()));