diff options
author | Sanjoy Das <sanjoy@google.com> | 2018-08-31 16:28:11 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-31 16:41:20 -0700 |
commit | 97f4d4016fc4a1eac5313bc9753aaeae118734a4 (patch) | |
tree | 98ff08921e18f4f85280aeb609fe2ef07d224006 /tensorflow/compiler/xla/service/cpu/ir_emitter.h | |
parent | b1341d049a74b7054450a9641b1c186f23df501f (diff) |
CHECK that the thread locality of the call matches thread locality of the callee
PiperOrigin-RevId: 211162384
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/ir_emitter.h')
-rw-r--r-- | tensorflow/compiler/xla/service/cpu/ir_emitter.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/ir_emitter.h b/tensorflow/compiler/xla/service/cpu/ir_emitter.h index 9cb8162327..6efd7fd001 100644 --- a/tensorflow/compiler/xla/service/cpu/ir_emitter.h +++ b/tensorflow/compiler/xla/service/cpu/ir_emitter.h @@ -571,6 +571,9 @@ class IrEmitter : public DfsHloVisitorWithDefault, tensorflow::gtl::FlatMap<BufferAllocation::Index, llvm::Constant*> constant_buffer_to_global_; + std::vector<const HloComputation*> thread_local_computations_; + std::vector<const HloComputation*> global_computations_; + TF_DISALLOW_COPY_AND_ASSIGN(IrEmitter); }; |