diff --git a/tensorflow/core/common_runtime/gpu/gpu_device.cc b/tensorflow/core/common_runtime/gpu/gpu_device.cc index c9696cf2307..ffe842d2f33 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_device.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_device.cc @@ -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_; } diff --git a/third_party/eigen3/workspace.bzl b/third_party/eigen3/workspace.bzl index a74dcd03ef6..73b1fe5e525 100644 --- a/third_party/eigen3/workspace.bzl +++ b/third_party/eigen3/workspace.bzl @@ -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",