Fixed compilation macro name issue

main
Kenneth Jao 2 years ago
parent 917c95d70c
commit b9002c19b7
  1. 4
      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();

Loading…
Cancel
Save