OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update Eigen to: 3d9051ea84

Browse Source 
Additional minor changes to account for API change in `TensorDeviceGpu` - static variables replaced
by a safer singleton class.

PiperOrigin-RevId: 374792818
Change-Id: I69a6bff974787302968f7ee306bec576b7c5c600
This commit is contained in:
Antonio Sanchez
2021-05-19 21:34:11 -07:00
committed by TensorFlower Gardener
parent 72088b0c4e
commit f4ccc2ca47
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/common_runtime/gpu/gpu_device.cc
View File

@@ -115,9 +115,7 @@ using se::rocm::ScopedActivateExecutorContext;
class EigenGpuStreamDevice : public ::Eigen::StreamInterface {
public:
EigenGpuStreamDevice()
: scratch_(nullptr), semaphore_(nullptr), context_(nullptr) {
Eigen::initializeDeviceProp();
}
: scratch_(nullptr), semaphore_(nullptr), context_(nullptr) {}
~EigenGpuStreamDevice() override {}
void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream,
TfDeviceId tf_device_id, ::tensorflow::Allocator* alloc,
@@ -135,7 +133,7 @@ class EigenGpuStreamDevice : public ::Eigen::StreamInterface {
PlatformDeviceId platform_device_id;
TF_CHECK_OK(
GpuIdManager::TfToPlatformDeviceId(tf_device_id, &platform_device_id));
device_prop_ = &Eigen::m_deviceProperties[platform_device_id.value()];
device_prop_ = &Eigen::GetGpuDeviceProperties(platform_device_id.value());
}
const gpuStream_t& stream() const override { return *stream_; }

4
third_party/eigen3/workspace.bzl vendored
View File

@@ -6,8 +6,8 @@ def repo():
"""Imports Eigen."""
# Attention: tools parse and update these lines.
EIGEN_COMMIT = "fcb5106c6e16599eeabadf7d82a465d52229698f"
EIGEN_SHA256 = "1d8c683cb8aa814499bd0254250937bebdc33146691a5d3c6b941af9d0303ea7"
EIGEN_COMMIT = "3d9051ea84a5089b277c88dac456b3b1576bfa7f"
EIGEN_SHA256 = "1e473b94966cd3084c6df5b4fe0612ea681ac0ae6624bd151950262c29ebb917"
tf_http_archive(
name = "eigen_archive",