diff --git a/projects/rccl/test/CorrectnessTest.hpp b/projects/rccl/test/CorrectnessTest.hpp index 33f5b56058..8894ee3767 100644 --- a/projects/rccl/test/CorrectnessTest.hpp +++ b/projects/rccl/test/CorrectnessTest.hpp @@ -553,9 +553,10 @@ dropback: envString = 0; numTokens = 0; + setenv("RCCL_TEST_ENV_VARS", "ENABLE", 1); if (strcmp(envVals, "")) { // enable RCCL env vars testing - setenv("RCCL_TEST_ENV_VARS", "ENABLE", 1); + envString = strdup(envVals); tokens[numTokens] = strtok(envString, "=, "); numTokens++; @@ -802,9 +803,9 @@ dropback: envString = 0; numTokens = 0; + setenv("RCCL_TEST_ENV_VARS", "ENABLE", 1); if (strcmp(envVals, "")) { // enable RCCL env vars testing - setenv("RCCL_TEST_ENV_VARS", "ENABLE", 1); envString = strdup(envVals); tokens[numTokens] = strtok(envString, "=, "); numTokens++;