From 98734577689133f8868926f9f46704a29a6c71b4 Mon Sep 17 00:00:00 2001 From: Sicheng Stephen Jia Date: Tue, 10 Dec 2024 11:41:14 -0500 Subject: [PATCH] Use libtorch target in generated operator tests/benchmarks instead of custom op lib target Differential Revision: D66964157 Pull Request resolved: https://github.com/pytorch/executorch/pull/7245 --- backends/vulkan/test/op_tests/targets.bzl | 22 ++-------------------- 1 file changed, 2 insertions(+), 20 deletions(-) diff --git a/backends/vulkan/test/op_tests/targets.bzl b/backends/vulkan/test/op_tests/targets.bzl index 4770dc5708..ab55d5beea 100644 --- a/backends/vulkan/test/op_tests/targets.bzl +++ b/backends/vulkan/test/op_tests/targets.bzl @@ -82,24 +82,6 @@ def define_common_targets(is_fbcode = False): default_outs = ["."], ) - pt_operator_library( - name = "all_aten_ops", - check_decl = False, - include_all_operators = True, - ) - - runtime.cxx_library( - name = "all_aten_ops_lib", - srcs = [], - define_static_target = False, - exported_deps = get_pt_ops_deps( - name = "pt_ops_full", - deps = [ - ":all_aten_ops", - ], - ), - ) - runtime.cxx_binary( name = "compute_graph_op_tests_bin", srcs = [ @@ -109,7 +91,7 @@ def define_common_targets(is_fbcode = False): deps = [ "//third-party/googletest:gtest_main", "//executorch/backends/vulkan:vulkan_graph_runtime", - ":all_aten_ops_lib", + runtime.external_dep_location("libtorch"), ], ) @@ -125,7 +107,7 @@ def define_common_targets(is_fbcode = False): deps = [ "//third-party/benchmark:benchmark", "//executorch/backends/vulkan:vulkan_graph_runtime", - ":all_aten_ops_lib", + runtime.external_dep_location("libtorch"), ], )