Merge pull request #27 from markkimsal/fixes/implicit-func-warning

fix: quiet compilation warning about implicit functions
NumPower · Jul 25, 2023 · 2dd212e · 2dd212e
2 parents f90cb2b + c50c0f0
commit 2dd212e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/ndarray.c b/src/ndarray.c
@@ -963,13 +963,15 @@ NDArray_Broadcast(NDArray *a, NDArray *b) {
                     rtn_p = rtn_p + (sizeof(float) * NDArray_SHAPE(src)[0]);
                 }
             }
+#ifdef HAVE_CUBLAS
             if (NDArray_DEVICE(dst) == NDARRAY_DEVICE_GPU) {
                 for (i = 0; i < NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 2]; i++) {
                     NDArray_VMEMCPY_D2D(NDArray_DATA(src), rtn_p,
                                         sizeof(float) * NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 1]);
                     rtn_p = rtn_p + (sizeof(float) * NDArray_SHAPE(src)[0]);
                 }
             }
+#endif
         }
     }
     int j;
@@ -982,6 +984,7 @@ NDArray_Broadcast(NDArray *a, NDArray *b) {
                     }
                 }
             }
+#ifdef HAVE_CUBLAS
             if (NDArray_DEVICE(dst) == NDARRAY_DEVICE_GPU) {
                 for (i = 0; i < NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 2]; i++) {
                     for (j = 0; j < NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 1]; j++) {
@@ -990,8 +993,8 @@ NDArray_Broadcast(NDArray *a, NDArray *b) {
                         NDArray_VMEMCPY_D2D(tmp_p, rtn_p, sizeof(float));
                     }
                 }
-                NDArray_Print(rtn,0);
             }
+#endif
         }
         if (NDArray_SHAPE(src)[NDArray_NDIM(dst) - 1] == NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 2]) {
             if (NDArray_DEVICE(dst) == NDARRAY_DEVICE_CPU) {
@@ -1001,6 +1004,7 @@ NDArray_Broadcast(NDArray *a, NDArray *b) {
                     rtn_p = rtn_p + (sizeof(float) * NDArray_SHAPE(src)[NDArray_NDIM(dst) - 1]);
                 }
             }
+#ifdef HAVE_CUBLAS
             if (NDArray_DEVICE(dst) == NDARRAY_DEVICE_GPU) {
                 for (i = 0; i < NDArray_SHAPE(dst)[NDArray_NDIM(dst) - 2]; i++) {
                     NDArray_VMEMCPY_D2D(NDArray_DATA(src), rtn_p,
@@ -1009,6 +1013,7 @@ NDArray_Broadcast(NDArray *a, NDArray *b) {
                                       (i * NDArray_STRIDES(rtn)[NDArray_NDIM(rtn) - 2] / NDArray_ELSIZE(rtn)) + j);
                 }
             }
+#endif
         }
     }
     return rtn;