Skip to content

Commit

Permalink
Use libtorch target in generated operator tests/benchmarks instead of…
Browse files Browse the repository at this point in the history
… custom op lib target

Differential Revision: D66964157

Pull Request resolved: #7245
  • Loading branch information
SS-JIA authored Dec 10, 2024
1 parent f6a87ac commit 9873457
Showing 1 changed file with 2 additions and 20 deletions.
22 changes: 2 additions & 20 deletions backends/vulkan/test/op_tests/targets.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [
Expand All @@ -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"),
],
)

Expand All @@ -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"),
],
)

Expand Down

0 comments on commit 9873457

Please sign in to comment.