From d65bd79b9fd4f00f4f46301d1841358a3af48d99 Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Wed, 14 Oct 2026 18:41:28 -0700 Subject: [PATCH 1/3] extend tests/others/test_cli_commands.py::TestRunCommand::test_compile_arg and tests/others/test_cli_commands.py::TestRunCommand::test_attention_backend_arg to other device Signed-off-by: Wang, Yi A --- tests/others/test_cli_commands.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/others/test_cli_commands.py b/tests/others/test_cli_commands.py index ba30cfb4a92b..f4b7715c5e0c 100644 --- a/tests/others/test_cli_commands.py +++ b/tests/others/test_cli_commands.py @@ -50,7 +50,7 @@ require_accelerator, require_kernels_version_greater_or_equal, require_torch_accelerator, - require_torch_gpu, + require_torch, ) @@ -228,7 +228,7 @@ def test_vae_slicing_arg(self): pipeline = _load_pipeline(args) assert pipeline.vae.use_slicing is True - @require_torch_gpu + @require_torch def test_compile_arg(self): args = self._parse_run_argv(["--compile"]) pipeline = _load_pipeline(args) @@ -242,7 +242,7 @@ def test_compile_arg(self): assert compiled_blocks assert all(m._compiled_call_impl is not None for m in compiled_blocks) - @require_torch_gpu + @require_torch # `--attention-backend` only exposes Hub-hosted kernels, all of which need `kernels>=0.12`. @require_kernels_version_greater_or_equal("0.12") def test_attention_backend_arg(self): From 93006e0e791a8401aa692291c210e9a640eb79af Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Wed, 14 Oct 2026 19:17:05 -0700 Subject: [PATCH 2/3] update xpu expectation Signed-off-by: Wang, Yi A --- .../marigold/test_marigold_intrinsics.py | 49 ++++++------------- 1 file changed, 14 insertions(+), 35 deletions(-) diff --git a/tests/pipelines/marigold/test_marigold_intrinsics.py b/tests/pipelines/marigold/test_marigold_intrinsics.py index 945bfe1bf448..4b27c908d679 100644 --- a/tests/pipelines/marigold/test_marigold_intrinsics.py +++ b/tests/pipelines/marigold/test_marigold_intrinsics.py @@ -388,10 +388,7 @@ def test_marigold_intrinsics_einstein_f32_accelerator_G0_S1_P768_E1_B1_M1(self): expected_slices = Expectations( { ("xpu", 5): np.array( - [0.62477, 0.62261, 0.62027, 0.62281, 0.62241, 0.62271, 0.62184, 0.62479, 0.63226] - ), - ("xpu", 3): np.array( - [0.62127, 0.61906, 0.61687, 0.61946, 0.61903, 0.61961, 0.61808, 0.62099, 0.62894] + [0.61482, 0.61291, 0.61132, 0.61320, 0.61308, 0.61398, 0.61246, 0.61555, 0.62334] ), (None, None): np.array( [0.62127, 0.61906, 0.61687, 0.61946, 0.61903, 0.61961, 0.61808, 0.62099, 0.62894] @@ -414,10 +411,7 @@ def test_marigold_intrinsics_einstein_f16_accelerator_G0_S1_P768_E1_B1_M1(self): expected_slices = Expectations( { ("xpu", 5): np.array( - [0.62451, 0.62256, 0.62012, 0.62256, 0.62207, 0.62256, 0.62158, 0.62451, 0.63184] - ), - ("xpu", 3): np.array( - [0.62109, 0.61914, 0.61719, 0.61963, 0.61914, 0.61963, 0.61816, 0.62109, 0.62891] + [0.61475, 0.61279, 0.61133, 0.61328, 0.61279, 0.61377, 0.61230, 0.61523, 0.62305] ), (None, None): np.array( [0.62109, 0.61914, 0.61719, 0.61963, 0.61914, 0.61963, 0.61816, 0.62109, 0.62891] @@ -440,9 +434,8 @@ def test_marigold_intrinsics_einstein_f16_accelerator_G2024_S1_P768_E1_B1_M1(sel expected_slices = Expectations( { ("xpu", 5): np.array( - [0.63330, 0.63135, 0.62793, 0.63184, 0.63135, 0.63135, 0.63037, 0.63379, 0.64160] + [0.63770, 0.63574, 0.63330, 0.63623, 0.63623, 0.63672, 0.63428, 0.63770, 0.64697] ), - ("xpu", 3): np.array([0.64111, 0.63916, 0.63623, 0.63965, 0.63916, 0.63965, 0.6377, 0.64062, 0.64941]), (None, None): np.array( [0.64111, 0.63916, 0.63623, 0.63965, 0.63916, 0.63965, 0.6377, 0.64062, 0.64941] ), @@ -463,9 +456,8 @@ def test_marigold_intrinsics_einstein_f16_accelerator_G2024_S1_P768_E1_B1_M1(sel def test_marigold_intrinsics_einstein_f16_accelerator_G0_S2_P768_E1_B1_M1(self): expected_slices = Expectations( { - ("xpu", 5): np.array([0.61475, 0.61328, 0.6123, 0.61426, 0.61328, 0.61475, 0.61475, 0.61621, 0.62402]), - ("xpu", 3): np.array( - [0.60254, 0.60059, 0.59961, 0.60156, 0.60107, 0.60205, 0.60254, 0.60449, 0.61133] + ("xpu", 5): np.array( + [0.60400, 0.60254, 0.60156, 0.60352, 0.60303, 0.60400, 0.60449, 0.60693, 0.61426] ), (None, None): np.array( [0.60254, 0.60059, 0.59961, 0.60156, 0.60107, 0.60205, 0.60254, 0.60449, 0.61133] @@ -500,30 +492,17 @@ def test_marigold_intrinsics_einstein_f16_accelerator_G0_S1_P512_E1_B1_M1(self): def test_marigold_intrinsics_einstein_f16_accelerator_G0_S1_P768_E3_B1_M1(self): expected_slices = Expectations( { - ("xpu", 3): np.array( - [ - 0.62655, - 0.62477, - 0.62161, - 0.62452, - 0.62454, - 0.62454, - 0.62255, - 0.62647, - 0.63379, - ] - ), ("xpu", 5): np.array( [ - 0.62354, - 0.62158, - 0.61963, - 0.62207, - 0.62158, - 0.62207, - 0.62109, - 0.62354, - 0.63135, + 0.61816, + 0.61621, + 0.61426, + 0.61621, + 0.61621, + 0.61670, + 0.61523, + 0.61816, + 0.62598, ] ), ("cuda", 7): np.array( From c934dcefe6302ed150fd45c1dc629044f5c66cbf Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Wed, 14 Oct 2026 19:54:34 -0700 Subject: [PATCH 3/3] update Signed-off-by: Wang, Yi A --- .../marigold/test_marigold_intrinsics.py | 25 +++++-------------- 1 file changed, 6 insertions(+), 19 deletions(-) diff --git a/tests/pipelines/marigold/test_marigold_intrinsics.py b/tests/pipelines/marigold/test_marigold_intrinsics.py index 4b27c908d679..b17fa7b07fa9 100644 --- a/tests/pipelines/marigold/test_marigold_intrinsics.py +++ b/tests/pipelines/marigold/test_marigold_intrinsics.py @@ -536,30 +536,17 @@ def test_marigold_intrinsics_einstein_f16_accelerator_G0_S1_P768_E3_B1_M1(self): def test_marigold_intrinsics_einstein_f16_accelerator_G0_S1_P768_E4_B2_M1(self): expected_slices = Expectations( { - ("xpu", 3): np.array( - [ - 0.62988, - 0.62792, - 0.62548, - 0.62841, - 0.62792, - 0.62792, - 0.62646, - 0.62939, - 0.63721, - ] - ), ("xpu", 5): np.array( [ - 0.62207, 0.62012, + 0.61816, + 0.61621, 0.61865, + 0.61816, + 0.61914, + 0.61768, 0.62061, - 0.62061, - 0.62158, - 0.62012, - 0.62305, - 0.63086, + 0.62842, ] ), ("cuda", 7): np.array(