Skip gpu_hlo_runner_test if input is not provided (#71)

ROCm · Jan 7, 2025 · 6a03a2d · 6a03a2d
1 parent 0a9ee68
commit 6a03a2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/xla/service/gpu/tests/BUILD b/xla/service/gpu/tests/BUILD
@@ -224,6 +224,7 @@ xla_test(
     name = "gpu_hlo_runner_test",
     srcs = ["gpu_hlo_runner_test.cc"],
     backends = ["gpu"],
+    tags=["no_oss"],
     deps = [
         ":gpu_codegen_test",
         "//xla:error_spec",

diff --git a/xla/service/gpu/tests/gpu_hlo_runner_test.cc b/xla/service/gpu/tests/gpu_hlo_runner_test.cc
@@ -41,7 +41,9 @@ class HloRunnerTest : public GpuCodegenTest {};
 TEST_F(HloRunnerTest, RunSingle) {
 
   std::ifstream ifs("input.hlo");
-  ASSERT_TRUE(ifs.good());
+  if(ifs.fail()) {
+    GTEST_SKIP() << "No input HLO file provided!";
+  }
 
   std::stringstream buffer;
   buffer << ifs.rdbuf();