Add missing HAVE_GPU guards.

Signed-off-by: Josh Romero <joshr@nvidia.com>
horovod · Jun 9, 2021 · 104485b · 104485b
1 parent 73e9077
commit 104485b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/horovod/common/common.h b/horovod/common/common.h
@@ -179,10 +179,12 @@ inline std::string CommunicatorName(Communicator comm) {
 
 struct Event {
   Event() = default;
+#if HAVE_GPU
   Event(std::shared_ptr<gpuEvent_t> event, gpuStream_t stream) :
     event(event), stream(stream) {};
   std::shared_ptr<gpuEvent_t> event;
   gpuStream_t stream = nullptr;
+#endif
 };
 
 
@@ -275,11 +277,13 @@ class ReadyEventList {
     return ready_events_.size();
   }
 
+#if HAVE_GPU
   void PushEventsToSet(std::unordered_set<gpuEvent_t>& event_set) {
     for (auto& e : ready_events_) {
       event_set.insert(e->event());
     }
   }
+#endif
 
   ~ReadyEventList() = default;