enable gpu test in internal ci

Differential Revision: D89829958

Pull Request resolved: #16404

Author: Gasoonjia

backends/cuda/runtime/shims/tests/targets.bzl

@@ -19,6 +19,10 @@ def cuda_shim_cpp_unittest(name):
         external_deps = [
             ("cuda", None, "cuda-lazy"),
         ],
+        keep_gpu_sections = True,
+        remote_execution = re_test_utils.remote_execution(
+            platform = "gpu-remote-execution",
+        ),
     )
 
 def define_common_targets():

backends/cuda/runtime/tests/targets.bzl

@@ -1,4 +1,5 @@
 load("@fbcode_macros//build_defs:cpp_unittest.bzl", "cpp_unittest")
+load("@fbcode_macros//build_defs/lib:re_test_utils.bzl", "re_test_utils")
 
 def cuda_runtime_cpp_unittest(name):
     cpp_unittest(
@@ -15,6 +16,10 @@ def cuda_runtime_cpp_unittest(name):
         external_deps = [
             ("cuda", None, "cuda-lazy"),
         ],
+        keep_gpu_sections = True,
+        remote_execution = re_test_utils.remote_execution(
+            platform = "gpu-remote-execution",
+        ),
     )
 
 def define_common_targets():