Skip to content

Commit

Permalink
_pywrap_profiler_plugin seems have been messed up somewhere, as a cyc…
Browse files Browse the repository at this point in the history
…le of errors
  • Loading branch information
cj401-amd committed Dec 20, 2024
1 parent f22b6c3 commit 2277278
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion tensorflow/python/profiler/internal/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -200,6 +200,15 @@ tsl_pybind_extension(
"//tensorflow/python/profiler:__pkg__",
]),
deps = [
"@local_xla//xla/tsl/platform/default:resource",
"@local_tsl//tsl/platform:random",
"@local_xla//xla/tsl/lib/monitoring:collection_registry",
"//tensorflow/core:framework_internal_impl",
"@local_xla//xla/tsl/lib/histogram:histogram",
"@local_xla//xla/tsl/lib/monitoring:sampler",
"//tensorflow/core/platform:platform_strings",
"//tensorflow/core/lib/core:arena",
"//tensorflow/core/common_runtime/gpu:gpu_id_impl", # weekly-sync:24-12-10
"//tensorflow/core:protos_all_cc_impl",
"//tensorflow/core/framework:attr_value_proto_cc_impl",
"//tensorflow/core/framework:op",
Expand All @@ -209,6 +218,10 @@ tsl_pybind_extension(
"//tensorflow/core/profiler/convert:xplane_to_tools_data",
"//tensorflow/python/lib/core:py_exception_registry",
"//tensorflow/python/lib/core:pybind11_status",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_protobuf//:protobuf",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/profiler/protobuf:profiler_analysis_proto_cc_impl",
Expand Down Expand Up @@ -256,4 +269,4 @@ tsl_pybind_extension(
"@local_tsl//tsl/profiler/lib:profiler_factory_impl",
"@local_tsl//tsl/profiler/lib:profiler_session_impl",
]),
)
)

0 comments on commit 2277278

Please sign in to comment.