pytorch
7d2b2dec - [Pytoch][Vulkan] Register `run_conv1d_context` (#122172)

Commit
1 year ago
[Pytoch][Vulkan] Register `run_conv1d_context` (#122172) Summary: We have rewritten `conv1d` as `create_conv1d_context` and `run_conv1d_context` to enable prepack of `weight` and `bias`. We have registered `create_conv1d_context` but not `run_conv1d_context`. We add the registration in this diff. Test Plan: ``` [luwei@devbig439.ftw3 /data/users/luwei/fbsource (f89a7de33)]$ LD_LIBRARY_PATH=third-party/swiftshader/lib/linux-x64/ buck run fbcode/mode/dev-nosan //xplat/caffe2:pt_vulkan_api_test_bin -- --gtest_filter="*conv1d*" Using additional configuration options from /home/luwei/.buckconfig.d/experiments_from_buck_start Recommended: For faster builds try buck2: replace 'buck' with 'buck2' NOTE: buck-out/ has changed: look for files in fbsource/buck-out/v2/ 'buck2 build --show-output //xplat/caffe2:pt_vulkan_api_test_bin' will print the new output paths. If you are building in fbsource//xplat and have questions, post in 'Cross Platform Dev Discussions': https://fb.workplace.com/groups/xplat.qa Targets matching .buckconfig buck2.supported_projects: {'//xplat/caffe2:pt_vulkan_api_test_bin': '//xplat'} To suppress this warning: touch ~/.config/.dont_hint_buck2 Building: finished in 0.1 sec (100%) 394/394 jobs, 0/394 updated Total time: 0.2 sec BUILD SUCCEEDED Running main() from third-party/googletest/1.14.0/googletest/googletest/src/gtest_main.cc Note: Google Test filter = *conv1d* [==========] Running 2 tests from 1 test suite. [----------] Global test environment set-up. [----------] 2 tests from VulkanAPITest [ RUN ] VulkanAPITest.conv1d_simple [ OK ] VulkanAPITest.conv1d_simple (208 ms) [ RUN ] VulkanAPITest.conv1d [ OK ] VulkanAPITest.conv1d (81 ms) [----------] 2 tests from VulkanAPITest (289 ms total) [----------] Global test environment tear-down [==========] 2 tests from 1 test suite ran. (289 ms total) [ PASSED ] 2 tests. ``` full test result ``` ... [----------] 427 tests from VulkanAPITest (22583 ms total) [----------] Global test environment tear-down [==========] 427 tests from 1 test suite ran. (22583 ms total) [ PASSED ] 426 tests. [ SKIPPED ] 1 test, listed below: [ SKIPPED ] VulkanAPITest.querypool_flushed_shader_log YOU HAVE 11 DISABLED TESTS ``` Differential Revision: D55052816 Pull Request resolved: https://github.com/pytorch/pytorch/pull/122172 Approved by: https://github.com/nathanaelsee
Author
Committer
Parents
Loading