Update elemwise_binary_broadcast_op_basic.cu

This fix apache#18170 refer to the fix: apache#18357
chinakook · Jul 20, 2020 · 2f0e0e8 · 2f0e0e8
1 parent 146b49e
commit 2f0e0e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/elemwise_binary_broadcast_op_basic.cu b/src/operator/tensor/elemwise_binary_broadcast_op_basic.cu
@@ -23,7 +23,7 @@
  * \brief GPU Implementation of basic functions for elementwise binary broadcast operator.
  */
 #include "./elemwise_unary_op.h"
-#include "./elemwise_binary_op.h"
+#include "./elemwise_binary_op-inl.h"
 #include "./elemwise_binary_broadcast_op.h"
 
 namespace mxnet {