We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 258a443 commit aa1a6daCopy full SHA for aa1a6da
torch/csrc/jit/codegen/cuda/arith.cpp
@@ -455,6 +455,8 @@ TensorView* reductionOp(
455
456
TORCH_CHECK(tv->nDims() > 0, "Tried to reduce a 0-dim tensor");
457
458
+ TORCH_CHECK(axes.size() > 0, "No reduction axis specified");
459
+
460
std::vector<unsigned int> uint_axes;
461
for (int axis : axes) {
462
if (axis < 0)
0 commit comments