diff --git a/backends/xnnpack/test/tester/TARGETS b/backends/xnnpack/test/tester/TARGETS index 72dc24a75a6..21fc329cef6 100644 --- a/backends/xnnpack/test/tester/TARGETS +++ b/backends/xnnpack/test/tester/TARGETS @@ -17,6 +17,7 @@ runtime.python_library( "//caffe2:torch", "//executorch/backends/xnnpack/partition:xnnpack_partitioner", "//executorch/backends/xnnpack/utils:xnnpack_utils", + "//executorch/devtools/visualization:visualization", "//executorch/exir:lib", "//executorch/exir/_serialize:lib", "//executorch/exir/backend:backend_api", diff --git a/devtools/visualization/TARGETS b/devtools/visualization/TARGETS new file mode 100644 index 00000000000..cb4e8b3cdc1 --- /dev/null +++ b/devtools/visualization/TARGETS @@ -0,0 +1,32 @@ +# Any targets that should be shared between fbcode and xplat must be defined in +# targets.bzl. This file can contain fbcode-only targets. +load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") + +oncall("executorch") + +runtime.python_library( + name = "visualization", + srcs = [ + "__init__.py", + "visualization_utils.py", + ], + visibility = [ + "//executorch/...", + "@EXECUTORCH_CLIENTS", + ], + deps = [ + "//caffe2:torch", + "//executorch/exir:lib", + "//executorch/exir/_serialize:lib", + ], +) + +runtime.python_library( + name = "test", + srcs = [ + "visualization_utils_test.py", + ], + deps = [ + ":visualization", + ] +)