From 3e16f6bca368e1e1d00e66c9628ab8985eb9eed8 Mon Sep 17 00:00:00 2001
From: Wonchan Lee <wonchanl@nvidia.com>
Date: Tue, 22 Mar 2022 14:36:15 -0700
Subject: [PATCH] Set the right flags for Thrust to fix the build issue

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

diff --git a/src/Makefile b/src/Makefile
index f9ca14ca06..d66b3d56bc 100644
--- a/src/Makefile
+++ b/src/Makefile
@@ -43,6 +43,7 @@ LIBNAME = libcunumeric
 
 CC_FLAGS ?=
 CC_FLAGS += -I. -I$(OPENBLAS_PATH)/include -I$(TBLIS_PATH)/include -I$(THRUST_PATH)
+CC_FLAGS += -DTHRUST_HOST_SYSTEM=THRUST_HOST_SYSTEM_CPP -DTHRUST_DEVICE_SYSTEM=THRUST_DEVICE_SYSTEM_OMP
 
 LD_FLAGS ?=
 LD_FLAGS += -L$(OPENBLAS_PATH)/lib -l$(OPENBLAS_LIBNAME) -Wl,-rpath,$(OPENBLAS_PATH)/lib