diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2017-02-22 16:38:20 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-22 17:01:38 -0800 |
| commit | 91d2cc5cb4cb2d2463e3ed7ea323fc627c4a2098 (patch) | |
| tree | 417c10abcb592629b5c1cb27d609f54a71561555 /tensorflow/compiler/jit/defs.h | |
| parent | eabdedd1ae712e587546f4b460954e1b80dad92d (diff) | |
Avoid merging adjacent XLA compilations from different scopes/functions
This is part 2 of the bugfix. It implements the xla chain breaking mechanism
based on different coloring of the graph (as represented by different XlaScope
strings). In part 1, we modified both experimental_jit_scope and Defun to
mark their ops as having different XlaScopes, so this is the final change that
actually enables the fusion breaking.
Also fixed a bug where xla_enabled was not True for rnn_cell_tests (the XLA benchmarks in that test were not actually being run with xla enabled).
Change: 148286731
Diffstat (limited to 'tensorflow/compiler/jit/defs.h')
| -rw-r--r-- | tensorflow/compiler/jit/defs.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/jit/defs.h b/tensorflow/compiler/jit/defs.h index ddc830cb77..a3aabc949d 100644 --- a/tensorflow/compiler/jit/defs.h +++ b/tensorflow/compiler/jit/defs.h @@ -23,6 +23,7 @@ namespace tensorflow { // Name of attribute used to tag operators for compilation with XLA extern const char* const kXlaCompileAttr; // "_XlaCompile" +extern const char* const kXlaScopeAttr; // "_XlaScope" } // namespace tensorflow |