From 934e6f22c5dcc161f2a65e4fcf1d71b145442c93 Mon Sep 17 00:00:00 2001
From: chinthysl <chin.dev.acc@gmail.com>
Date: Tue, 25 Jun 2024 09:35:24 +0000
Subject: [PATCH] Fix missing compiler flag

---
 Makefile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Makefile b/Makefile
index 1a2fa02ad..92f37ae70 100644
--- a/Makefile
+++ b/Makefile
@@ -198,6 +198,7 @@ else
       $(info ✗ Multi-GPU on CUDA on Darwin is not supported, skipping NCCL support)
     else ifeq ($(shell dpkg -l | grep -q nccl && echo "exists"), exists)
       $(info ✓ NCCL found, OK to train with multiple GPUs)
+      NVCC_FLAGS += -DMULTI_GPU
       NVCC_LDLIBS += -lnccl
     else
       $(info ✗ NCCL is not found, disabling multi-GPU support)