kern.cu 413 字节
Newer Older
M
Megvii Engine Team 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
#include "./kern.cuh"

namespace megdnn {
namespace cuda {
#define cb(_dtype)                                             \
    INST_RUN_ELEMWISE(                                         \
            MaskedFillScalarKernOp<DTypeTrait<_dtype>::ctype>, \
            DTypeTrait<_dtype>::ctype, 1);
MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
cb(::megdnn::dtype::Bool)

#undef cb
}  // namespace cuda
}  // namespace megdnn