update

sxjscience · Dec 28, 2017 · e99ffd0 · e99ffd0
1 parent 399ba02
commit e99ffd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/indexing_op.h b/src/operator/tensor/indexing_op.h
@@ -1154,7 +1154,7 @@ struct scatter_nd_acc {
       offset += strides[j] * static_cast<int>(indices[j*N + i]);
     }
     for (int j = 0; j < K; ++j) {
-#if __CUDA__ 
+#ifdef __CUDACC__
       atomicAdd(out + (offset + j), data[i * K + j]);
 #else
       out[offset + j] += data[i * K + j];