diff options
author | 2017-11-09 20:45:39 -0800 | |
---|---|---|
committer | 2017-11-10 16:14:41 -0800 | |
commit | 64d9aa1ace99c66f20b65532f633acb34ee3c057 (patch) | |
tree | 42589268a62815c66093ad7185eb507b1562f9fb /tensorflow/compiler/xla/service/hlo_runner.cc | |
parent | 685f604f63a30a8162d8762e9d8d22f171dca85e (diff) |
Add bfloat support to XLA.
This is necessary in providing bfloat support in GPU backend.
RELNOTES: bfloat support is now added to XLA infra.
PiperOrigin-RevId: 175252067
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_runner.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_runner.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_runner.cc b/tensorflow/compiler/xla/service/hlo_runner.cc index f463e57d99..158fb9a546 100644 --- a/tensorflow/compiler/xla/service/hlo_runner.cc +++ b/tensorflow/compiler/xla/service/hlo_runner.cc @@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ +#define EIGEN_USE_THREADS #include "tensorflow/compiler/xla/service/hlo_runner.h" @@ -19,8 +20,6 @@ limitations under the License. #include <string> #include <utility> -#define EIGEN_USE_THREADS - #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/compiler/xla/layout_util.h" #include "tensorflow/compiler/xla/ptr_util.h" |