[CI] Upgrade unity image tag to 20240917-153130-9f281758 (#17410)

* upgrade docker image to `20240917-153130-9f281758`

* fix dynamo test case

* building torch requires c++ 17

* temporary skip jax gpu tests due to XlaRuntimeError

Author: mshr-h

ci/jenkins/unity_jenkinsfile.groovy

@@ -30,14 +30,14 @@
 import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
 
 // NOTE: these lines are scanned by docker/dev_common.sh. Please update the regex as needed. -->
-ci_lint = 'tlcpack/ci-lint:20240105-165030-51bdaec6'
-ci_gpu = 'tlcpack/ci-gpu:20240105-165030-51bdaec6'
-ci_cpu = 'tlcpack/ci-cpu:20240105-165030-51bdaec6'
+ci_lint = 'tlcpack/ci_lint:20240917-153130-9f281758'
+ci_gpu = 'tlcpack/ci_gpu:20240917-153130-9f281758'
+ci_cpu = 'tlcpack/ci_cpu:20240917-153130-9f281758'
 ci_wasm = 'tlcpack/ci-wasm:v0.72'
 ci_i386 = 'tlcpack/ci-i386:v0.75'
 ci_qemu = 'tlcpack/ci-qemu:v0.11'
 ci_arm = 'tlcpack/ci-arm:v0.08'
-ci_hexagon = 'tlcpack/ci-hexagon:20240105-165030-51bdaec6'
+ci_hexagon = 'tlcpack/ci_hexagon:20240917-153130-9f281758'
 // <--- End of regex-scanned config.
 
 // Parameters to allow overriding (in Jenkins UI), the images

src/contrib/msc/plugin/torch_codegen.cc

@@ -219,7 +219,7 @@ void TorchPluginCodeGen::CodeGenCmake(const std::set<String>& devices) {
   flags.Set("PLUGIN_SUPPORT_TORCH", "");
   CodeGenPreCmake(devices, flags);
   stack_.line()
-      .line("set(CMAKE_CXX_STANDARD 14)")
+      .line("set(CMAKE_CXX_STANDARD 17)")
       .line("list(APPEND CMAKE_PREFIX_PATH \"" + config()->torch_prefix + "\")")
       .line("find_package(Torch REQUIRED)");
   Array<String> includes, libs;

tests/python/relax/test_frontend_dynamo.py

@@ -223,7 +223,7 @@ def subgraph_1(
         ) -> R.Tensor((10,), dtype="float32"):
             # block 0
             with R.dataflow():
-                lv5: R.Tensor((10,), dtype="float32") = R.multiply(inp_11, inp_01)
+                lv5: R.Tensor((10,), dtype="float32") = R.multiply(inp_01, inp_11)
                 gv1: R.Tensor((10,), dtype="float32") = lv5
                 R.output(gv1)
             return gv1

tests/python/relax/test_frontend_stablehlo.py

@@ -196,6 +196,10 @@ def main(
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_unary():
     import jax
 
@@ -229,6 +233,10 @@ def _round(x):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_binary():
     import jax
 
@@ -250,6 +258,10 @@ def fn(x, y):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_const():
     import jax
 
@@ -260,6 +272,10 @@ def fn(x):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_maximum():
     import jax
     import jax.numpy as jnp
@@ -271,6 +287,10 @@ def fn(x, y):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_minimum():
     import jax
     import jax.numpy as jnp
@@ -282,6 +302,10 @@ def fn(x, y):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_reduce():
     import jax
     import jax.numpy as jnp
@@ -293,6 +317,10 @@ def fn(x):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_reduce_window():
     import jax
     from flax import linen as nn
@@ -304,6 +332,10 @@ def fn(x):
 
 
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
+# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_dot_general():
     import jax
 
@@ -314,8 +346,10 @@ def fn(x, y):
     check_correctness(jax.jit(fn), input_shapes)
 
 
-@pytest.mark.skip()
 @tvm.testing.requires_gpu
+@pytest.mark.skip(
+    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
+)
 # TODO(yongwww): fix flaky error of "invalid device ordinal"
 def test_conv():
     import jax