diff --git a/xla/stream_executor/rocm/rocm_dnn.cc b/xla/stream_executor/rocm/rocm_dnn.cc
index aacc62fda1d33..c83d4fac52af3 100644
--- a/xla/stream_executor/rocm/rocm_dnn.cc
+++ b/xla/stream_executor/rocm/rocm_dnn.cc
@@ -22,6 +22,7 @@ limitations under the License.
 #include <memory>
 
 #include "absl/algorithm/container.h"
+#include "absl/base/optimization.h"
 #include "absl/base/thread_annotations.h"
 #include "absl/strings/str_cat.h"
 #include "absl/strings/str_format.h"
@@ -1966,7 +1967,7 @@ class MixinBase<void> {};
 }  // namespace
 
 #define RETURN_IF_MIOPEN_ERROR(STATUS, ...)                                   \
-  if (!SE_PREDICT_TRUE((STATUS) == miopenStatusSuccess)) {                    \
+  if (!ABSL_PREDICT_TRUE((STATUS) == miopenStatusSuccess)) {                  \
     std::string error_msg = absl::StrCat(ToString(STATUS), " ", __VA_ARGS__); \
     SetFailure(::absl::UnknownError(error_msg));                              \
     LOG(ERROR) << error_msg;                                                  \