Fixing simplectraninit and fastinit tests by addressing the new colltrace enable.

Summary:
The latest changes enable the new colltrace by default. As a result, tests that expects the legacy colltrace are failing during the colltrace check.

In this diff I have updated a those chekings to keep these tests alive.

For reference: https://fb.workplace.com/groups/752760876991686/permalink/985386930395745/

Reviewed By: YulunW

Differential Revision: D85580342

fbshipit-source-id: a892f6da417d4953482328b57c115a7b4a04cfdc

Author: mlunar-meta

comms/ncclx/v2_27/meta/tests/FastInitTest.cc

@@ -35,7 +35,7 @@ void validateCtranInitialization(
   ASSERT_NE(nullptr, comm->ctranComm_);
   ASSERT_NE(nullptr, comm->ctranComm_->statex_);
   ASSERT_NE(nullptr, comm->ctranComm_->bootstrap_);
-  ASSERT_NE(nullptr, comm->ctranComm_->collTrace_);
+  ASSERT_NE(nullptr, comm->ctranComm_->colltraceNew_);
   ASSERT_NE(nullptr, comm->ctranComm_->ctran_);
   EXPECT_TRUE(ctranInitialized(comm->ctranComm_.get()));
   EXPECT_EQ(comm->commHash, comm->ctranComm_->statex_->commHash());
@@ -431,6 +431,7 @@ INSTANTIATE_TEST_SUITE_P(
         {"NCCL_FASTINIT_MODE", "ring_hybrid"},
         {"NCCL_CTRAN_ENABLE", "1"},
         {"NCCL_COLLTRACE", "trace"},
+        {"NCCL_COLLTRACE_USE_NEW_COLLTRACE", "1"},
     })),
     [](const testing::TestParamInfo<NcclxBaseTestFixture::ParamType>& info) {
       // generate test-name for a given NcclxEnvs

comms/ncclx/v2_27/meta/tests/SimpleCtranInitTest.cc

@@ -13,6 +13,7 @@ class SimpleCtranInitTest : public ::testing::Test {
   void SetUp() override {
     setenv("NCCL_CTRAN_ENABLE", "1", 1);
     setenv("NCCL_COLLTRACE", "trace", 1);
+    setenv("NCCL_COLLTRACE_USE_NEW_COLLTRACE", "1", 1);
     setenv("NCCL_USE_MEM_CACHE", "1", 1);
     setenv("NCCL_LAZY_SETUP_CHANNELS", "1", 1);
     setenv("NCCL_RUNTIME_CONNECT", "1", 1);
@@ -39,7 +40,7 @@ TEST_F(SimpleCtranInitTest, VerifyCtranCommStructures) {
   ASSERT_NE(nullptr, comm->ctranComm_->statex_);
   ASSERT_NE(nullptr, comm->ctranComm_->bootstrap_);
   ASSERT_NE(nullptr, comm->ctranComm_->memCache_);
-  ASSERT_NE(nullptr, comm->ctranComm_->collTrace_);
+  ASSERT_NE(nullptr, comm->ctranComm_->colltraceNew_);
   ASSERT_NE(nullptr, comm->ctranComm_->ctran_);
   EXPECT_TRUE(ctranInitialized(comm->ctranComm_.get()));
   EXPECT_EQ(comm->commHash, comm->ctranComm_->statex_->commHash());