Skip to content

Commit

Permalink
Unify how cpp test is being run (#8391)
Browse files Browse the repository at this point in the history
  • Loading branch information
JackCaoG authored Nov 19, 2024
1 parent fbed916 commit 2ec2264
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 25 deletions.
2 changes: 1 addition & 1 deletion BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,6 @@ test_suite(
"//test/cpp:test_tensor",
"//test/cpp:test_xla_sharding",
"//torch_xla/csrc/runtime:pjrt_computation_client_test",
"//torch_xla/csrc/runtime:ifrt_computation_client_test",
# "//torch_xla/csrc/runtime:ifrt_computation_client_test",
],
)
48 changes: 24 additions & 24 deletions torch_xla/csrc/runtime/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -497,27 +497,27 @@ ptxla_cc_test(
],
)

ptxla_cc_test(
name = "ifrt_computation_client_test",
srcs = ["ifrt_computation_client_test.cc"],
deps = [
":computation_client",
":ifrt_computation_client",
":tensor_source",
"@xla//xla/tsl/lib/core:status_test_util",
"@tsl//tsl/platform:env",
"@tsl//tsl/platform:errors",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:test",
"@tsl//tsl/platform:test_main",
"@xla//xla:literal",
"@xla//xla:literal_util",
"@xla//xla:shape_util",
"@xla//xla:status",
"@xla//xla:statusor",
"@xla//xla/client:xla_builder",
"@xla//xla/client:xla_computation",
"@xla//xla/tests:literal_test_util",
"@xla//xla/tools:hlo_module_loader",
],
)
# ptxla_cc_test(
# name = "ifrt_computation_client_test",
# srcs = ["ifrt_computation_client_test.cc"],
# deps = [
# ":computation_client",
# ":ifrt_computation_client",
# ":tensor_source",
# "@xla//xla/tsl/lib/core:status_test_util",
# "@tsl//tsl/platform:env",
# "@tsl//tsl/platform:errors",
# "@tsl//tsl/platform:logging",
# "@tsl//tsl/platform:test",
# "@tsl//tsl/platform:test_main",
# "@xla//xla:literal",
# "@xla//xla:literal_util",
# "@xla//xla:shape_util",
# "@xla//xla:status",
# "@xla//xla:statusor",
# "@xla//xla/client:xla_builder",
# "@xla//xla/client:xla_computation",
# "@xla//xla/tests:literal_test_util",
# "@xla//xla/tools:hlo_module_loader",
# ],
# )

0 comments on commit 2ec2264

Please sign in to comment.