Skip to content

GitLab

Commit 46ee5b45 authored by Graylatzhou's avatar Graylatzhou
Browse files

issue/183 已解决冲突部分

parent 03edef48
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/infiniop/elementwise/cuda/elementwise_cuda.cuh
View file @ 46ee5b45
...@@ -359,8 +359,7 @@ private: ...@@ -359,8 +359,7 @@ private:
d_input_shapes, d_input_strides, stream)); d_input_shapes, d_input_strides, stream));
dim3 blockDims(std::min(BLOCK_SIZE, static_cast<uint32_t>(internal->maxThreadsPerBlock()))); dim3 blockDims(std::min(BLOCK_SIZE, static_cast<uint32_t>(internal->maxThreadsPerBlock())));
dim3 gridDims(std::min(CEIL_DIV(output_size, blockDims.x), static_cast<size_t>(internal->gridSizeX()))); dim3 gridDims(std::min(uint32_t(CEIL_DIV(output_size, blockDims.x)), static_cast<uint32_t>(internal->gridSizeX())));
size_t step = gridDims.x * blockDims.x; size_t step = gridDims.x * blockDims.x;
for (size_t i = 0; i < output_size; i += step) { for (size_t i = 0; i < output_size; i += step) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment