pytorch
dc00bb51 - [Vulkan][TCC] Add tests for conv2d prepack context (#88316)

Commit
2 years ago
[Vulkan][TCC] Add tests for conv2d prepack context (#88316) Summary: Implement Vulkan tests for the create/run context functions in Convolution.cpp, their transposed versions and their backwards compatible versions: - create_conv2d_context - run_conv2d_context - create_tconv2d_context - run_tconv2d_context - conv2d_clamp_prepack - conv2d_clamp_run Test Plan: On Mac ``` cd ~/fbsource buck run -c pt.vulkan_full_precision=1 //xplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 ``` On Android ``` cd ~/fbsource buck build -c ndk.custom_libcxx=false -c pt.enable_qpl=0 -c pt.vulkan_full_precision=1 //xplat/caffe2:pt_vulkan_api_test_binAndroid\#android-arm64 --show-output adb push buck-out/gen/xplat/caffe2/pt_vulkan_api_test_binAndroid\#android-arm64 /data/local/tmp/vulkan_api_test adb shell "/data/local/tmp/vulkan_api_test" ``` Reviewed By: salilsdesai Differential Revision: D40935343 Pull Request resolved: https://github.com/pytorch/pytorch/pull/88316 Approved by: https://github.com/salilsdesai
Committer
Parents
Loading