blob: a4d513a4d08c8914ebc9d15089935fbdcf17996b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
diff --git a/tensorflow/stream_executor/rocm/rocblas_wrapper.h b/tensorflow/stream_executor/rocm/rocblas_wrapper.h
--- a/tensorflow/stream_executor/rocm/rocblas_wrapper.h (revision 9685e1ca56f7ba1d31b84d3dda51d50c92b36ae7)
+++ b/tensorflow/stream_executor/rocm/rocblas_wrapper.h (date 1663176000544)
@@ -20,7 +20,7 @@
#ifndef TENSORFLOW_STREAM_EXECUTOR_ROCM_ROCBLAS_WRAPPER_H_
#define TENSORFLOW_STREAM_EXECUTOR_ROCM_ROCBLAS_WRAPPER_H_
-#include "rocm/include/rocblas.h"
+#include "rocm/include/rocblas/rocblas.h"
#include "tensorflow/stream_executor/gpu/gpu_activation.h"
#include "tensorflow/stream_executor/lib/env.h"
#include "tensorflow/stream_executor/platform/dso_loader.h"
diff --git a/tensorflow/stream_executor/rocm/rocm_blas.h b/tensorflow/stream_executor/rocm/rocm_blas.h
--- a/tensorflow/stream_executor/rocm/rocm_blas.h (revision 9685e1ca56f7ba1d31b84d3dda51d50c92b36ae7)
+++ b/tensorflow/stream_executor/rocm/rocm_blas.h (date 1663175994438)
@@ -22,7 +22,7 @@
#include "absl/base/thread_annotations.h"
#include "absl/synchronization/mutex.h"
-#include "rocm/include/rocblas.h"
+#include "rocm/include/rocblas/rocblas.h"
#include "tensorflow/stream_executor/blas.h"
#include "tensorflow/stream_executor/platform/port.h"
#include "tensorflow/stream_executor/plugin_registry.h"
diff --git a/tensorflow/core/util/gpu_solvers.h b/tensorflow/core/util/gpu_solvers.h
--- a/tensorflow/core/util/gpu_solvers.h (revision 9685e1ca56f7ba1d31b84d3dda51d50c92b36ae7)
+++ b/tensorflow/core/util/gpu_solvers.h (date 1663176032850)
@@ -33,7 +33,7 @@
#else
#include "rocm/rocm_config.h"
#include "rocm/include/hip/hip_complex.h"
-#include "rocm/include/rocblas.h"
+#include "rocm/include/rocblas/rocblas.h"
#include "tensorflow/stream_executor/blas.h"
#if TF_ROCM_VERSION >= 40500
#include "tensorflow/stream_executor/rocm/hipsolver_wrapper.h"
|