pytorch

5c395520 - Add linear quantize function to custom ops (#111148)

Add linear quantize function to custom ops (#111148) Summary: Add linear quantize for vulkan to custom ops so it can be used from a model. Test Plan: buck2 run --target-platforms ovr_config//platform/macos:arm64-fbsource -c pt.vulkan_full_precision=1 //xplat/caffe2/fb/custom_ops/vulkan_quantized:pt_vulkan_quantized_test_binAppleMac\#macosx-arm64 [ OK ] VulkanAPITest.convert_qconv2d_context (135 ms) [ RUN ] VulkanAPITest.linear_2d [ OK ] VulkanAPITest.linear_2d (4 ms) [----------] 2 tests from VulkanAPITest (139 ms total) [----------] Global test environment tear-down [==========] 2 tests from 1 test suite ran. (139 ms total) [ PASSED ] 2 tests. ############################################################## buck2 build --target-platforms ovr_config//platform/macos:arm64-fbsource //xplat/caffe2:pt_vulkan_quantized_api_test_binAppleMac\#macosx-arm64 -c pt.vulkan_full_precision=1 --show-output" buck-out//v2/gen/fbsource/xplat/caffe2/pt_vulkan_quantized_api_test_binAppleMac [ OK ] VulkanAPITest.conv2d_pw_quantized_prepack_random_params_int8_int32 (11 ms) [ RUN ] VulkanAPITest.linear_2d_flat [ OK ] VulkanAPITest.linear_2d_flat (4 ms) [ RUN ] VulkanAPITest.linear_2d_small [ OK ] VulkanAPITest.linear_2d_small (1 ms) [ RUN ] VulkanAPITest.linear_2d_large [ OK ] VulkanAPITest.linear_2d_large (1 ms) [ RUN ] VulkanAPITest.linear_3d_flat [ OK ] VulkanAPITest.linear_3d_flat (2 ms) [ RUN ] VulkanAPITest.linear_3d_small [ OK ] VulkanAPITest.linear_3d_small (2 ms) [ RUN ] VulkanAPITest.linear_3d_large [ OK ] VulkanAPITest.linear_3d_large (1 ms) [ RUN ] VulkanAPITest.linear_4d_flat [ OK ] VulkanAPITest.linear_4d_flat (1 ms) [ RUN ] VulkanAPITest.linear_4d_small [ OK ] VulkanAPITest.linear_4d_small (1 ms) [ RUN ] VulkanAPITest.linear_4d_large [ OK ] VulkanAPITest.linear_4d_large (1 ms) [ RUN ] VulkanAPITest.linear_custom [ OK ] VulkanAPITest.linear_custom (0 ms) [----------] 76 tests from VulkanAPITest (1811 ms total) [----------] Global test environment tear-down [==========] 76 tests from 1 test suite ran. (1811 ms total) [ PASSED ] 76 tests. YOU HAVE 8 DISABLED TESTS ############################################################## buck2 run --target-platforms ovr_configplatform/macos:arm64-fbsourcexplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 -c pt.vulkan_full_precision=1 [----------] Global test environment tear-down [==========] 346 tests from 1 test suite ran. (5648 ms total) [ PASSED ] 345 tests. [ SKIPPED ] 1 test, listed below: [ SKIPPED ] VulkanAPITest.querypool_flushed_shader_log YOU HAVE 5 DISABLED TESTS Reviewed By: manuelcandales Differential Revision: D49609985 Pull Request resolved: https://github.com/pytorch/pytorch/pull/111148 Approved by: https://github.com/yipjustin