diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD
index a753d1e68a6445..adb79a867e3a76 100644
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@@ -1976,7 +1976,7 @@ tf_xla_py_strict_test(
 
 tf_xla_py_strict_test(
     name = "sort_ops_test",
-    size = "medium",
+    size = "large",
     srcs = ["sort_ops_test.py"],
     shard_count = 50,
     # Times out in fastbuild mode.