"docs/git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "b21fdd537350c8e652f263fca9439c7fa323ac7e"
Unverified Commit 52203fa4 authored by Attila Dusnoki's avatar Attila Dusnoki Committed by GitHub
Browse files

Fix wrong size check when axes not present for slice (#2270)

parent 685168f5
......@@ -137,7 +137,7 @@ struct parse_slice : op_parser<parse_slice>
sd.always_insert(args.at(0));
// If axes arg is not given, the default is all of them.
if(sd.op.axes.empty() and sd.op_args.size() < 3)
if(sd.op.axes.empty() and sd.op_args.size() <= 3)
{
std::vector<int64_t> axes(args[0]->get_shape().ndim());
std::iota(axes.begin(), axes.end(), int64_t{0});
......
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