diff --git a/tensorflow/core/kernels/cubin_headers/BUILD b/tensorflow/core/kernels/cubin_headers/BUILD
index b8ba164fbc3..ec8b44050db 100644
--- a/tensorflow/core/kernels/cubin_headers/BUILD
+++ b/tensorflow/core/kernels/cubin_headers/BUILD
@@ -37,4 +37,5 @@ gen_kernel_library(
         "f32",
         "f64",
     ],
+    unroll_factors = "4",
 )