提交 717e1252 编写于 作者: L Luo Tao

fix some warning

上级 ba4322f4
...@@ -62,7 +62,7 @@ struct ForRange<CUDADeviceContext> { ...@@ -62,7 +62,7 @@ struct ForRange<CUDADeviceContext> {
template <typename Function> template <typename Function>
inline void operator()(Function func) const { inline void operator()(Function func) const {
constexpr size_t num_threads = 1024; constexpr int num_threads = 1024;
int block_size = limit_ <= num_threads ? limit_ : num_threads; int block_size = limit_ <= num_threads ? limit_ : num_threads;
int grid_size = (limit_ + num_threads - 1) / num_threads; int grid_size = (limit_ + num_threads - 1) / num_threads;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册