• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_cc_test")
3load(
4    "//tensorflow/core/platform:build_config_root.bzl",
5    "tf_cuda_tests_tags",
6)
7
8package(
9    default_visibility = [
10        "//learning/brain:__subpackages__",
11        "//tensorflow/compiler:__subpackages__",
12    ],
13    licenses = ["notice"],
14)
15
16cc_library(
17    name = "raw_api_test_lib",
18    testonly = 1,
19    srcs = [
20        "raw_api_test.cc",
21    ],
22    deps = [
23        "//tensorflow/cc:cc_ops",
24        "//tensorflow/cc:client_session",
25        "//tensorflow/cc:ops",
26        "//tensorflow/cc:scope",
27        "//tensorflow/compiler/tf2xla:common",
28        "//tensorflow/compiler/tf2xla:xla_compiler",
29        "//tensorflow/compiler/xla:literal",
30        "//tensorflow/compiler/xla:literal_util",
31        "//tensorflow/compiler/xla:shape_util",
32        "//tensorflow/compiler/xla:xla_data_proto_cc",
33        "//tensorflow/compiler/xla/client:client_library",
34        "//tensorflow/compiler/xla/client:local_client",
35        "//tensorflow/compiler/xla/client:xla_builder",
36        "//tensorflow/compiler/xla/client:xla_computation",
37        "//tensorflow/compiler/xla/client/lib:arithmetic",
38        "//tensorflow/compiler/xla/client/lib:constants",
39        "//tensorflow/compiler/xla/service:platform_util",
40        "//tensorflow/compiler/xrt:xrt_proto_cc",
41        "//tensorflow/compiler/xrt:xrt_server",
42        "//tensorflow/compiler/xrt/cc:xrt_ops",
43        "//tensorflow/core:core_cpu_internal",
44        "//tensorflow/core:framework",
45        "//tensorflow/core:framework_internal",
46        "//tensorflow/core:lib",
47        "//tensorflow/core:tensorflow_opensource",
48        "//tensorflow/core:test",
49        "//tensorflow/core:test_main",
50        "@com_google_absl//absl/strings",
51    ],
52)
53
54tf_cc_test(
55    name = "raw_api_test_cpu",
56    size = "medium",
57    srcs = [],
58    args = [
59        "--xla_test_device=XLA_CPU",
60        "--xla_platform=CPU",
61    ],
62    deps = [
63        ":raw_api_test_lib",
64        "//tensorflow/compiler/jit:xla_cpu_device",
65    ],
66)
67
68tf_cuda_cc_test(
69    name = "raw_api_test_gpu",
70    size = "medium",
71    srcs = [],
72    args = [
73        "--xla_test_device=XLA_GPU",
74        "--xla_platform=GPU",
75    ],
76    tags = tf_cuda_tests_tags() + [
77        "no_cuda_asan",  # TODO(b/171319142): re-enable.
78    ],
79    deps = [
80        ":raw_api_test_lib",
81        "//tensorflow/compiler/jit:xla_gpu_device",
82    ],
83)
84