Skip to content

Commit 2785b4e

Browse files
committed
Update on "[8/n][ET-VK] Support staging any 8-bit texture"
"bitw8" = bit width 8, which is equivalent to 8-bit. We use "bitw8" as the name since shader compilation disallows names starting with a digit. Changes follow from #4485 to support `texture2d` and support `uint8`, respectively. Differential Revision: [D63918659](https://our.internmc.facebook.com/intern/diff/D63918659/) [ghstack-poisoned]
2 parents f0315db + 9706821 commit 2785b4e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

backends/vulkan/test/vulkan_compute_api_test.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2365,7 +2365,8 @@ void run_from_gpu_test(
23652365

23662366
if (dtype == vkapi::kChar &&
23672367
!context()->adapter_ptr()->has_full_int8_buffers_support()) {
2368-
record_bitw8_image_to_nchw_nobitw8buffer_op(context(), vten, staging_buffer);
2368+
record_bitw8_image_to_nchw_nobitw8buffer_op(
2369+
context(), vten, staging_buffer);
23692370
} else {
23702371
record_image_to_nchw_op(context(), vten, staging_buffer.buffer());
23712372
}
@@ -2412,7 +2413,8 @@ void round_trip_test(
24122413
// Copy data in and out of the tensor
24132414
if (dtype == vkapi::kChar &&
24142415
!context()->adapter_ptr()->has_full_int8_buffers_support()) {
2415-
record_bitw8_image_to_nchw_nobitw8buffer_op(context(), vten, staging_buffer_out);
2416+
record_bitw8_image_to_nchw_nobitw8buffer_op(
2417+
context(), vten, staging_buffer_out);
24162418
} else {
24172419
record_image_to_nchw_op(context(), vten, staging_buffer_out.buffer());
24182420
}

0 commit comments

Comments
 (0)