Skip to content

Commit bc2b805

Browse files
committed
Fix bugs
2 parents 2d371bc + 937a297 commit bc2b805

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

paddle/fluid/operators/elementwise/elementwise_floordiv_op.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
See the License for the specific language governing permissions and
1313
limitations under the License. */
1414
#include "paddle/fluid/operators/elementwise/elementwise_floordiv_op.h"
15+
#include "paddle/fluid/operators/elementwise/elementwise_op_broadcast.cu.h"
16+
#include "paddle/fluid/operators/elementwise/elementwise_op_function.cu.h"
1517
#include "paddle/fluid/platform/float16.h"
1618

1719
namespace ops = paddle::operators;

paddle/fluid/operators/elementwise/elementwise_op_function.cu.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ struct CudaFloorDivFunctor {
261261
"InvalidArgumentError: Divide by zero"
262262
"encounterd in floor_divide.\n Please check!");
263263
#endif
264-
return static_cast<T>(std::trunc(args[0] / args[1]]));
264+
return static_cast<T>(std::trunc(args[0] / args[1]));
265265
}
266266
};
267267

0 commit comments

Comments
 (0)