Use the right include for gmock and gtest

PiperOrigin-RevId: 709043164
jax-ml · Dec 23, 2024 · 80b765a · 80b765a
1 parent 83e60a9
commit 80b765a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jaxlib/mosaic/dialect/gpu/mosaic_gpu_test.cc b/jaxlib/mosaic/dialect/gpu/mosaic_gpu_test.cc
@@ -19,8 +19,8 @@ limitations under the License.
 #include <optional>
 #include <string>
 
-#include "testing/base/public/gmock.h"
-#include "testing/base/public/gunit.h"
+#include <gmock/gmock.h>
+#include <gtest/gtest.h>
 #include "absl/container/flat_hash_set.h"
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"