Skip to content

Commit cb5d21a

Browse files
author
morelos
committed
Update on "[ET-VK][Ops] aten.var.dim from scratch implementation"
Created the var.dim operator (which functionally supports var) from scratch Differential Revision: [D75244137](https://our.internmc.facebook.com/intern/diff/D75244137/) [ghstack-poisoned]
2 parents 947e963 + 1e0c686 commit cb5d21a

File tree

1 file changed

+1
-1
lines changed
  • backends/vulkan/runtime/graph/ops/impl

1 file changed

+1
-1
lines changed

backends/vulkan/runtime/graph/ops/impl/Var.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ void add_var_buffer_node(
4242
reduce_dim = nchw_dim_to_whcn_dim(reduce_dim, ndim);
4343

4444
// Check that the concat dim is not the reduction dim, if the tensor has a
45-
// batch dim greater than 1.
45+
// batch dim greater than 1
4646
if (graph.dim_of(in) == 4 && graph.size_at<int>(0, in) > 1) {
4747
VK_CHECK_COND(graph.concat_dim_of(in) != reduce_dim);
4848
VK_CHECK_COND(graph.concat_dim_of(out) != reduce_dim);

0 commit comments

Comments
 (0)