Skip to content

Commit 69a97bf

Browse files
author
morelos
committed
Update on "[ET-VK][Ops] aten.var.dim in reduce"
Incorporated variance logic into reduce by adding additional logic Differential Revision: [D75247432](https://our.internmc.facebook.com/intern/diff/D75247432/) [ghstack-poisoned]
2 parents 8c22723 + b9d3e0e commit 69a97bf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/vulkan/test/op_tests/cases.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1120,10 +1120,10 @@ def get_reduce_inputs(is_softmax: bool = False, is_variance: bool = False):
11201120
# Alternating unbiased for comprehensive testing for variance.
11211121
# We also pass in a True for keep_dim to be consistent with other ops.
11221122
unbiased = (i % 2) == 0
1123-
test_cases.append((shape, dim, unbiased, True))
1123+
test_cases.append((shape, dim, unbiased, bool_arg))
11241124
else:
11251125
for i, (shape, dim) in enumerate(shapes_and_dims):
1126-
test_cases.append((shape, dim, True))
1126+
test_cases.append((shape, dim, bool_arg))
11271127

11281128
return test_cases
11291129

0 commit comments

Comments
 (0)