Skip to content

Commit 51a21c7

Browse files
committed
Update on "[ET-VK] Enable FP16 type in operators"
Differential Revision: [D56189470](https://our.internmc.facebook.com/intern/diff/D56189470/) [ghstack-poisoned]
2 parents fb2b637 + 4682b50 commit 51a21c7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backends/vulkan/test/vulkan_compute_api_test.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -815,6 +815,10 @@ void run_from_gpu_test(
815815
api::GPUMemoryLayout::TENSOR_CHANNELS_PACKED,
816816
api::ScalarType dtype = api::kFloat,
817817
api::StorageType storage_type = api::StorageType::TEXTURE_3D) {
818+
if (dtype == api::kHalf &&
819+
!api::context()->adapter_ptr()->has_16bit_storage()) {
820+
return;
821+
}
818822
vTensor vten =
819823
vTensor(api::context(), sizes, dtype, storage_type, memory_layout);
820824

0 commit comments

Comments
 (0)