update with VLOG(2) and fix a bug, new line at the EoF

Author: cj401-amd

xla/backends/profiler/gpu/rocm_collector.cc

@@ -621,15 +621,14 @@ std::vector<RocmTracerEvent> RocmTraceCollectorImpl::ApiActivityInfoExchange() {
 
     if (api_event == api_events_map_.end()) {
       api_event = auxiliary_api_events_map_.find(activity_event.correlation_id);
-    }
-
-    if (api_event == auxiliary_api_events_map_.end()) {
-      OnEventsDropped(
-          "An event from activity was discarded."
-          "Could not find the counterpart HIP API.",
-          activity_event.correlation_id);
-      PrintRocmTracerEvent(activity_event, ". Dropped!");
-      continue;
+      if (api_event == auxiliary_api_events_map_.end()) {
+        OnEventsDropped(
+            "An event from activity was discarded."
+            "Could not find the counterpart HIP API.",
+            activity_event.correlation_id);
+        PrintRocmTracerEvent(activity_event, ". Dropped!");
+        continue;
+      }
     }
 
     switch (activity_event.type) {
@@ -684,4 +683,4 @@ std::unique_ptr<RocmTraceCollector> CreateRocmCollector(
 }
 
 }  // namespace profiler
-}  // namespace xla
+}  // namespace xla

xla/backends/profiler/gpu/rocm_collector.h

@@ -178,7 +178,7 @@ class RocmTraceCollectorImpl : public RocmTraceCollector {
 
   void OnEventsDropped(const std::string& reason,
                        uint32_t correlation_id) override {
-    LOG(INFO) << "RocmTracerEvent dropped (correlation_id=" << correlation_id
+    VLOG(2) << "RocmTracerEvent dropped (correlation_id=" << correlation_id
               << ",) : " << reason << ".";
   }
 
@@ -217,4 +217,4 @@ std::unique_ptr<RocmTraceCollector> CreateRocmCollector(
 }  // namespace profiler
 }  // namespace xla
 
-#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_COLLECTOR_H_
+#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_COLLECTOR_H_

xla/backends/profiler/gpu/rocm_collector_test.cc

@@ -113,4 +113,4 @@ TEST(RocmCollectorTest, TestAddKernelEventAndExport) {
 
 }  // namespace test
 }  // namespace profiler
-}  // namespace xla
+}  // namespace xla

xla/backends/profiler/gpu/rocm_tracer.cc

@@ -261,7 +261,7 @@ void RocmTracer::MemcpyEvent(const rocprofiler_record_header_t* hdr,
       .async = false,
   };
 
-  LOG(INFO) << "copy bytes: " << trace_event->memcpy_info.num_bytes
+  VLOG(2) << "copy bytes: " << trace_event->memcpy_info.num_bytes
             << " stream: " << trace_event->stream_id << " src_id "
             << trace_event->device_id << " dst_id "
             << trace_event->memcpy_info.destination;
@@ -573,4 +573,4 @@ extern "C" rocprofiler_tool_configure_result_t* rocprofiler_configure(
 
 void __attribute__((constructor)) init_rocm_lib() {
   rocprofiler_force_configure(xla::profiler::rocprofiler_configure);
-}
+}

xla/backends/profiler/gpu/rocm_tracer.h

@@ -134,4 +134,4 @@ class RocmTracer {
 }  // namespace profiler
 }  // namespace xla
 
-#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_TRACER_H_
+#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_TRACER_H_

xla/backends/profiler/gpu/rocm_tracer_test.cc

@@ -118,4 +118,4 @@ TEST(RocmTracerTest, AnnotationMapWorks) {
 
 }  // namespace test
 }  // namespace profiler
-}  // namespace xla
+}  // namespace xla

xla/backends/profiler/gpu/rocm_tracer_utils.cc

@@ -97,5 +97,5 @@ absl::string_view AnnotationMap::LookUp(uint32_t correlation_id) {
   return it != map_.correlation_map.end() ? it->second : absl::string_view();
 }
 
-} // namespace profiler
-} // namespace xla
+}  // namespace profiler
+}  // namespace xla

xla/backends/profiler/gpu/rocm_tracer_utils.h

@@ -203,4 +203,4 @@ class AnnotationMap {
 }  // namespace profiler
 }  // namespace xla
 
-#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_UTILS_H_
+#endif  // XLA_BACKENDS_PROFILER_GPU_ROCM_UTILS_H_

xla/stream_executor/rocm/roctracer_wrapper.h

@@ -31,4 +31,4 @@ limitations under the License.
 #include <rocm/include/rocprofiler-sdk/cxx/name_info.hpp>
 #include <rocm/include/rocprofiler-sdk/rocprofiler.h>
 
-#endif  // XLA_STREAM_EXECUTOR_ROCM_ROCTRACER_WRAPPER_H_
+#endif  // XLA_STREAM_EXECUTOR_ROCM_ROCTRACER_WRAPPER_H_