Skip to content

Commit 7c3c9a9

Browse files
committed
Update on "[ET-VK] Add binary op support for height and width packing GPU layouts"
## Context Enable `binary_op` to support inputs that are `HEIGHT_PACKED` and `WIDTH_PACKED`. Differential Revision: [D55031044](https://our.internmc.facebook.com/intern/diff/D55031044/) [ghstack-poisoned]
2 parents 459dc0d + aaa48d2 commit 7c3c9a9

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

backends/vulkan/test/vulkan_compute_api_test.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -782,12 +782,12 @@ void run_from_gpu_test(
782782

783783
submit_to_gpu();
784784

785-
std::vector<T> data(staging_buffer.numel());
785+
std::vector<T> data_out(staging_buffer.numel());
786786
copy_staging_to_ptr(
787-
staging_buffer, data.data(), sizeof(float) * staging_buffer.numel());
787+
staging_buffer, data_out.data(), sizeof(float) * staging_buffer.numel());
788788

789789
for (int i = 0; i < vten.numel(); i++) {
790-
CHECK_VALUE(data, i, i);
790+
CHECK_VALUE(data_out, i, i);
791791
}
792792
}
793793

@@ -823,15 +823,15 @@ void run_to_gpu_test(
823823
submit_to_gpu();
824824

825825
// Extract data from output staging buffer
826-
std::vector<T> data(staging_buffer_out.numel());
826+
std::vector<T> data_out(staging_buffer_out.numel());
827827
copy_staging_to_ptr(
828828
staging_buffer_out,
829-
data.data(),
829+
data_out.data(),
830830
sizeof(float) * staging_buffer_out.numel());
831831

832832
// All indices should be equal to the input data
833833
for (int i = 0; i < vten.numel(); i++) {
834-
CHECK_VALUE(data, i, i);
834+
CHECK_VALUE(data_out, i, i);
835835
}
836836
}
837837

0 commit comments

Comments
 (0)