This CL adds a filter mask to BFCAllocator::AddTraceMe so we can coll…

…ect only memory TraceMe events.

PiperOrigin-RevId: 700477323

tsl/profiler/protobuf/profiler_options.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package tensorflow;
 
-// Next ID: 11
+// Next ID: 12
 message ProfileOptions {
   // Some default value of option are not proto3 default value. Use this version
   // to determine if we should use default option value instead of proto3
@@ -63,6 +63,16 @@ message ProfileOptions {
 
   // Directory to save profile data to. No-op when empty.
   string repository_path = 10;
+
+  message TraceOptions {
+    // Filter mask for TraceMe events. If the traceme_filter_mask is set, a
+    // TraceMe event will be recorded if it passes the filter.
+    // Only lowest 32 bits of the mask are used. The higher 32 bits are reserved
+    // and won't be applied if set.
+    uint64 host_traceme_filter_mask = 1;
+  }
+
+  TraceOptions trace_options = 11;
 }
 
 // Options for remote profiler session manager.