update

sxjscience · sxjscience · commit e99ffd075832 · 2017-12-27T16:57:59.000-08:00
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];

Original file line number	Diff line number	Diff line change
`@@ -1154,7 +1154,7 @@ struct scatter_nd_acc {`
`1154`	`1154`	`offset += strides[j] * static_cast<int>(indices[j*N + i]);`
`1155`	`1155`	`}`
`1156`	`1156`	`for (int j = 0; j < K; ++j) {`
`1157`		`-#if __CUDA__`
	`1157`	`+#ifdef __CUDACC__`
`1158`	`1158`	`atomicAdd(out + (offset + j), data[i * K + j]);`
`1159`	`1159`	`#else`
`1160`	`1160`	`out[offset + j] += data[i * K + j];`