From 1f0af9020657fa9825e74bc62338d95e7d3c0f5e Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 13 Feb 2024 14:03:43 -0800
Subject: [PATCH] Fix undefined symbol when nvtx is not enabled (#1082)

---
 src/init_nvtx.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/init_nvtx.cc b/src/init_nvtx.cc
index 44face6814..dbd54a51a3 100644
--- a/src/init_nvtx.cc
+++ b/src/init_nvtx.cc
@@ -11,6 +11,7 @@ static constexpr const nvtxPayloadEnum_t NvtxEnumRedSchema[] = {
 
 // Must be called before the first call to any reduction operation.
 void initNvtxRegisteredEnums() {
+#ifndef NVTX_NO_IMPL
   // Register schemas and strings
   constexpr const nvtxPayloadEnumAttr_t eAttr {
     .fieldMask = NVTX_PAYLOAD_ENUM_ATTR_ENTRIES | NVTX_PAYLOAD_ENUM_ATTR_NUM_ENTRIES |
@@ -23,4 +24,5 @@ void initNvtxRegisteredEnums() {
   };
 
   nvtxPayloadEnumRegister(nvtx3::domain::get<nccl_domain>(), &eAttr);
+#endif
 }