Automated Code Change

PiperOrigin-RevId: 704061736
Intel-tensorflow · Dec 8, 2024 · fb46636 · fb46636
1 parent ace1e90
commit fb46636
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/xla/python/profiler/BUILD b/xla/python/profiler/BUILD
@@ -40,6 +40,7 @@ tsl_pybind_extension(
         ":profile_data_lib",
         "@nanobind",
         "@tsl//tsl/platform:protobuf",
+        "@tsl//tsl/profiler/protobuf:xplane_proto_cc",
     ],
 )
 

diff --git a/xla/python/profiler/profile_data.cc b/xla/python/profiler/profile_data.cc
@@ -26,6 +26,7 @@ limitations under the License.
 #include "tsl/platform/env.h"
 #include "tsl/platform/logging.h"
 #include "tsl/platform/status.h"
+#include "tsl/profiler/protobuf/xplane.pb.h"
 
 namespace tensorflow::profiler::python {
 

diff --git a/xla/python/profiler/profile_data.h b/xla/python/profiler/profile_data.h
@@ -18,8 +18,10 @@ limitations under the License.
 
 #include <nanobind/nanobind.h>
 
+#include <cstddef>
 #include <cstdint>
 #include <functional>
+#include <iterator>
 #include <memory>
 #include <string>
 

diff --git a/xla/python/profiler/py_profile_data.cc b/xla/python/profiler/py_profile_data.cc
@@ -21,6 +21,7 @@ limitations under the License.
 
 #include "xla/python/profiler/profile_data.h"
 #include "tsl/platform/protobuf.h"
+#include "tsl/profiler/protobuf/xplane.pb.h"
 
 namespace {