pytorch

a2fd2c2b - [Pytorch] Add Vulkan support for aten::unsqueeze for 2d to 3d (#101719)

[Pytorch] Add Vulkan support for aten::unsqueeze for 2d to 3d (#101719) Summary: Unsqueeze operator: https://pytorch.org/docs/stable/generated/torch.unsqueeze.html#torch.unsqueeze Test Plan: Unsqueeze tests: https://www.internalfb.com/phabricator/paste/view/P738187802 ``` lfq@lfq-mbp fbsource % buck run --target-platforms ovr_config//platform/macos:arm64-fbsource //xplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 -c pt.vulkan_full_precision=1 -- --gtest_filter="*unsqueeze*" Downloaded 0/2 artifacts, 0.00 bytes, 100.0% cache miss (for updated rules) Building: finished in 15.0 sec (100%) 455/455 jobs, 2/455 updated Total time: 15.0 sec BUILD SUCCEEDED Running main() from xplat/third-party/gmock/googletest-1.12.1/googletest/src/gtest_main.cc Note: Google Test filter = *unsqueeze* [==========] Running 3 tests from 1 test suite. [----------] Global test environment set-up. [----------] 3 tests from VulkanAPITest [ RUN ] VulkanAPITest.unsqueeze_dim0 [ OK ] VulkanAPITest.unsqueeze_dim0 (96 ms) [ RUN ] VulkanAPITest.unsqueeze_dim1 [ OK ] VulkanAPITest.unsqueeze_dim1 (2 ms) [ RUN ] VulkanAPITest.unsqueeze_dim2 [ OK ] VulkanAPITest.unsqueeze_dim2 (3 ms) [----------] 3 tests from VulkanAPITest (101 ms total) [----------] Global test environment tear-down [==========] 3 tests from 1 test suite ran. (101 ms total) [ PASSED ] 3 tests. ``` All tests: buck run //xplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 https://www.internalfb.com/phabricator/paste/view/P738255852 Reviewed By: SS-JIA Differential Revision: D45893511 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101719 Approved by: https://github.com/SS-JIA