From b9002c19b7ecac9ce68e8868eb7c5966a4a5050b Mon Sep 17 00:00:00 2001
From: Kenneth Jao <ksjao2@illinois.edu>
Date: Fri, 2 Jun 2023 14:47:00 -0500
Subject: [PATCH] Fixed compilation macro name issue

---
 BLAS.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/BLAS.h b/BLAS.h
index 2eb17cc..e315b2a 100644
--- a/BLAS.h
+++ b/BLAS.h
@@ -138,7 +138,7 @@ template <typename T> class Batch {
 
         Array<T> batch = arr.reshaped({mBatchSize, mShape.rows(), mShape.cols()});
         for (uint32_t i = 0; i < mBatchSize; ++i) {
-#ifdef CUDA
+#ifdef CUDACC
             mBatch[i] = batch[i].dataDevice();
 #else
             mBatch[i] = batch[i].data();
@@ -154,7 +154,7 @@ template <typename T> class Batch {
     void add(const Array<T>& arr) {
         CT_ERROR(not arr.isView(), "Cannot add non-view Arrays");
         CT_ERROR_IF(mCount, ==, mBatchSize, "Batch is full, cannot add more arrays");
-#ifdef CUDA
+#ifdef CUDACC
         mBatch[mCount] = arr.dataDevice();
 #else
         mBatch[mCount] = arr.data();