Migrate array_elementwise_ops_test to always use PjRt for its test backend.

PiperOrigin-RevId: 720791217

Author: nvgrw

third_party/xla/xla/tests/BUILD

@@ -885,10 +885,14 @@ xla_test(
     name = "array_elementwise_ops_test",
     srcs = ["array_elementwise_ops_test.cc"],
     shard_count = 25,
-    tags = ["test_xla_cpu_no_thunks"],
+    tags = [
+        "test_migrated_to_hlo_runner_pjrt",
+        "test_xla_cpu_no_thunks",
+    ],
     deps = [
         ":client_library_test_runner_mixin",
-        ":hlo_test_base",
+        ":hlo_pjrt_interpreter_reference_mixin",
+        ":hlo_pjrt_test_base",
         ":test_macros_header",
         ":xla_internal_test_main",
         "//xla:array2d",

third_party/xla/xla/tests/array_elementwise_ops_test.cc

@@ -43,7 +43,8 @@ limitations under the License.
 #include "xla/literal_util.h"
 #include "xla/primitive_util.h"
 #include "xla/tests/client_library_test_runner_mixin.h"
-#include "xla/tests/hlo_test_base.h"
+#include "xla/tests/hlo_pjrt_interpreter_reference_mixin.h"
+#include "xla/tests/hlo_pjrt_test_base.h"
 #include "xla/tests/test_macros.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/tsl/platform/test.h"
@@ -104,7 +105,8 @@ void AddNegativeValuesMaybeRemoveZero(std::vector<T>& values) {
 }
 
 class ArrayElementwiseOpTest
-    : public ClientLibraryTestRunnerMixin<HloTestBase> {
+    : public ClientLibraryTestRunnerMixin<
+          HloPjRtInterpreterReferenceMixin<HloPjRtTestBase>> {
  public:
   static constexpr float kEpsF32 = std::numeric_limits<float>::epsilon();
   static constexpr double kEpsF64 = std::numeric_limits<double>::epsilon();
@@ -1339,7 +1341,8 @@ XLA_TEST_F(ArrayElementwiseOpTest, CompareEqF32s) {
 }
 
 template <typename T>
-class TotalOrderTest : public ClientLibraryTestRunnerMixin<HloTestBase> {
+class TotalOrderTest : public ClientLibraryTestRunnerMixin<
+                           HloPjRtInterpreterReferenceMixin<HloPjRtTestBase>> {
  public:
   void DoIt(ComparisonDirection direction) {
     this->SetFastMathDisabled(true);