Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "c0652d907b2ee39d010598ec4c5f7f287fa41947"
Commit de930090 authored by rusty1s's avatar rusty1s
Browse files

reset

parent 75dbe175
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
cpu/dim_apply.h
View file @ de930090
......@@ -17,7 +17,7 @@
auto TENSOR3##_stride = TENSOR3.stride(DIM); \
\
auto dims = TENSOR1.dim(); \
auto zeros = at::zeros(dims, torch::CPU(at::kLong)); \
auto zeros = at::zeros(torch::CPU(at::kLong), {dims}); \
auto counter = zeros.data<int64_t>(); \
bool has_finished = false; \
\
......@@ -76,7 +76,7 @@
auto TENSOR4##_stride = TENSOR4.stride(DIM); \
\
auto dims = TENSOR1.dim(); \
auto zeros = at::zeros(dims, torch::CPU(at::kLong)); \
auto zeros = at::zeros(torch::CPU(at::kLong), {dims}); \
auto counter = zeros.data<int64_t>(); \
bool has_finished = false; \
\
......
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