From 9a7288ccf576a25ab65e61246242dd9bb90345bd Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Mon, 9 Mar 2020 10:47:13 -0700
Subject: [PATCH] Add dot configuration as part of module configuration.

PiperOrigin-RevId: 299879284
Change-Id: I64932896316470cda7556a3618b32ac3556b3dac
---
 tensorflow/compiler/xla/service/hlo_module_config.h | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tensorflow/compiler/xla/service/hlo_module_config.h b/tensorflow/compiler/xla/service/hlo_module_config.h
index dee601d9e96..fa2dbc0fcad 100644
--- a/tensorflow/compiler/xla/service/hlo_module_config.h
+++ b/tensorflow/compiler/xla/service/hlo_module_config.h
@@ -183,6 +183,12 @@ class HloModuleConfig {
     return &fusion_config_;
   }
 
+  const std::vector<std::vector<int64>>& dot_config() const {
+    return dot_config_;
+  }
+
+  std::vector<std::vector<int64>>* mutable_dot_config() { return &dot_config_; }
+
  private:
   // If you add new members, be sure to update compilation_cache_key.
 
@@ -214,6 +220,8 @@ class HloModuleConfig {
       FusionConfigCollection::kOff;
 
   std::vector<std::vector<bool>> fusion_config_;
+
+  std::vector<std::vector<int64>> dot_config_;
 };
 
 }  // namespace xla