Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
remove check from kernel
Browse files Browse the repository at this point in the history
  • Loading branch information
apeforest committed Dec 26, 2019
1 parent 56ec207 commit 0ae7a41
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/operator/nn/dropout-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,6 @@ class DropoutOp {
uint8_t *mask_out,
const DType *input_data,
const real_t pkeep) {
CHECK_EQ(step & 7, 0);
RNG_KERNEL_LOOP(xpu, DType, id, gen, N, step, {
const real_t rand_num = static_cast<real_t>(genImpl.uniform());
// mask_out is set per bit position
Expand Down

0 comments on commit 0ae7a41

Please sign in to comment.