[Pytorch][Vulkan] aten::pow (#105550)
Summary:
Add support for [aten::pow](https://pytorch.org/docs/stable/generated/torch.pow.html#torch.pow) in [various forms](https://www.internalfb.com/code/fbsource/[c717e1fa980ed47c6580778dcfa49c21d3270a67]/xplat/caffe2/aten/src/ATen/native/native_functions.yaml?lines=9656%2C9670%2C9685%2C9693%2C9699):
|Not in-place| Base | Exp |
|--| -- | -- |
|pow| Tensor | Tensor |
|pow_tensor_scalar| Tensor | Scalar |
|pow_scalar_tensor| Scalar | Tensor |
|In-place| Base | Exp |
|--| -- | -- |
|pow_ | Tensor | Tensor |
|pow_tensor_scalar_| Tensor | Scalar |
Test Plan:
pow tests
```
[lfq@35771.od /data/sandcastle/boxes/fbsource (97d4bdf9e)]$ LD_LIBRARY_PATH=third-party/swiftshader/lib/linux-x64/ buck run fbcode/mode/dev-nosan //xplat/caffe2:pt_vulkan_api_test_bin -- --gtest_filter="*pow*"
Building: finished in 0.1 sec (100%) 329/329 jobs, 0/329 updated
Total time: 0.2 sec
BUILD SUCCEEDED
Running main() from xplat/third-party/gmock/googletest-1.12.1/googletest/src/gtest_main.cc
Note: Google Test filter = *pow*
[==========] Running 7 tests from 1 test suite.
[----------] Global test environment set-up.
[----------] 7 tests from VulkanAPITest
[ RUN ] VulkanAPITest.pow
[ OK ] VulkanAPITest.pow (255 ms)
[ RUN ] VulkanAPITest.pow_broadcast
[ OK ] VulkanAPITest.pow_broadcast (3 ms)
[ RUN ] VulkanAPITest.pow_
[ OK ] VulkanAPITest.pow_ (90 ms)
[ RUN ] VulkanAPITest.pow_broadcast_other_
[ OK ] VulkanAPITest.pow_broadcast_other_ (0 ms)
[ RUN ] VulkanAPITest.pow_tensor_scalar
[ OK ] VulkanAPITest.pow_tensor_scalar (57 ms)
[ RUN ] VulkanAPITest.pow_tensor_scalar_
[ OK ] VulkanAPITest.pow_tensor_scalar_ (83 ms)
[ RUN ] VulkanAPITest.pow_scalar_tensor
[ OK ] VulkanAPITest.pow_scalar_tensor (50 ms)
[----------] 7 tests from VulkanAPITest (542 ms total)
[----------] Global test environment tear-down
[==========] 7 tests from 1 test suite ran. (542 ms total)
[ PASSED ] 7 tests.
```
All tests
```
QueryPool is not available
[ SKIPPED ] VulkanAPITest.querypool_flushed_shader_log (0 ms)
[----------] 317 tests from VulkanAPITest (18448 ms total)
[----------] Global test environment tear-down
[==========] 317 tests from 1 test suite ran. (18448 ms total)
[ PASSED ] 316 tests.
[ SKIPPED ] 1 test, listed below:
[ SKIPPED ] VulkanAPITest.querypool_flushed_shader_log
```
clang-format on glsl and cpp files
Reviewed By: SS-JIA
Differential Revision: D46704167
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105550
Approved by: https://github.com/SS-JIA