1 files changed, 6 insertions, 0 deletions
diff --git a/spark-common/src/main/proto/spark/spark_sampler.proto b/spark-common/src/main/proto/spark/spark_sampler.proto
index 245da37..60cffac 100644
--- a/spark-common/src/main/proto/spark/spark_sampler.proto
+++ b/spark-common/src/main/proto/spark/spark_sampler.proto
@@ -32,6 +32,7 @@ message SamplerMetadata {
   int32 number_of_ticks = 12;
   map<string, SourceMetadata> sources = 13;
   map<string, string> extra_platform_metadata = 14;
+  SamplerMode sampler_mode = 15;
 
   message ThreadDumper {
     Type type = 1;
@@ -67,6 +68,11 @@ message SamplerMetadata {
     string name = 1;
     string version = 2;
   }
+
+  enum SamplerMode {
+    EXECUTION = 0;
+    ALLOCATION = 1;
+  }
 }
 
 message ThreadNode {