pytorch
17fec516 - [Vulkan] Test conv2d after division (#100910)

Commit
1 year ago
[Vulkan] Test conv2d after division (#100910) Summary: This tests running a conv2d with clamp after dividing the input tensor by another tensor. Both tensors have number channels = 3 (i.e. not a multiple of 4) and therefore, the channel dimension was padded. Hence, we are testing our divide-by-zero fix (D44392406) Test Plan: ``` buck run --target-platforms ovr_config//platform/macos:arm64-fbsource -c pt.vulkan_full_precision=1 //xplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 -- --gtest_filter="VulkanAPITest.conv2d_clamp_after_div" ``` Reviewed By: SS-JIA Differential Revision: D44550026 Pull Request resolved: https://github.com/pytorch/pytorch/pull/100910 Approved by: https://github.com/SS-JIA
Committer
Parents
Loading