diff --git a/python/test/unit/language/print_helper.py b/python/test/unit/language/print_helper.py index b89494be7e..becc2ab603 100644 --- a/python/test/unit/language/print_helper.py +++ b/python/test/unit/language/print_helper.py @@ -131,20 +131,15 @@ def test_print(func: str, data_type: str, device: str): else: assert f"Unknown kernel: {func}" - if device == "xpu": - # FIXME: remove trigger to get output from kernel - repr(x) - repr(y) + # Wait until driver complete all the jobs for the device_print, especially test_subprocess + # require this which captures stdout when child exits. + getattr(torch, device).synchronize() if func != "print_no_arg" and func != "no_arg_print" and func != "device_print_large" and \ func != "print_multiple_args" and func != "device_print_multiple_args" and \ func != "device_print_pointer" and func != "device_print_scalar": assert_close(y, x) - # Wait until driver complete all the jobs for the device_print, especially test_subprocess - # require this which captures stdout when child exits. - getattr(torch, device).synchronize() - if __name__ == "__main__": fn = globals()[sys.argv[1]] diff --git a/scripts/skiplist/a770/subprocess.txt b/scripts/skiplist/a770/subprocess.txt deleted file mode 100644 index bcf5e0079e..0000000000 --- a/scripts/skiplist/a770/subprocess.txt +++ /dev/null @@ -1,2 +0,0 @@ -# https://github.com/intel/intel-xpu-backend-for-triton/issues/1704 -test/unit/language/test_subprocess.py::test_print[device_print_uint-uint32] diff --git a/scripts/skiplist/conda/subprocess.txt b/scripts/skiplist/conda/subprocess.txt deleted file mode 100644 index 6f0ada5109..0000000000 --- a/scripts/skiplist/conda/subprocess.txt +++ /dev/null @@ -1,9 +0,0 @@ -# https://github.com/intel/intel-xpu-backend-for-triton/issues/800 -test/unit/language/test_subprocess.py::test_print[device_print-float16] -test/unit/language/test_subprocess.py::test_print[device_print-float32] -test/unit/language/test_subprocess.py::test_print[device_print-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float16] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float32] -# https://github.com/intel/intel-xpu-backend-for-triton/issues/1704 -test/unit/language/test_subprocess.py::test_print[device_print_uint-uint32] diff --git a/scripts/skiplist/default/subprocess.txt b/scripts/skiplist/default/subprocess.txt deleted file mode 100644 index 6f0ada5109..0000000000 --- a/scripts/skiplist/default/subprocess.txt +++ /dev/null @@ -1,9 +0,0 @@ -# https://github.com/intel/intel-xpu-backend-for-triton/issues/800 -test/unit/language/test_subprocess.py::test_print[device_print-float16] -test/unit/language/test_subprocess.py::test_print[device_print-float32] -test/unit/language/test_subprocess.py::test_print[device_print-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float16] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float32] -# https://github.com/intel/intel-xpu-backend-for-triton/issues/1704 -test/unit/language/test_subprocess.py::test_print[device_print_uint-uint32] diff --git a/scripts/skiplist/lts/subprocess.txt b/scripts/skiplist/lts/subprocess.txt deleted file mode 100644 index 6f0ada5109..0000000000 --- a/scripts/skiplist/lts/subprocess.txt +++ /dev/null @@ -1,9 +0,0 @@ -# https://github.com/intel/intel-xpu-backend-for-triton/issues/800 -test/unit/language/test_subprocess.py::test_print[device_print-float16] -test/unit/language/test_subprocess.py::test_print[device_print-float32] -test/unit/language/test_subprocess.py::test_print[device_print-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float16] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float32] -# https://github.com/intel/intel-xpu-backend-for-triton/issues/1704 -test/unit/language/test_subprocess.py::test_print[device_print_uint-uint32] diff --git a/scripts/skiplist/no-basekit/subprocess.txt b/scripts/skiplist/no-basekit/subprocess.txt deleted file mode 100644 index 6f0ada5109..0000000000 --- a/scripts/skiplist/no-basekit/subprocess.txt +++ /dev/null @@ -1,9 +0,0 @@ -# https://github.com/intel/intel-xpu-backend-for-triton/issues/800 -test/unit/language/test_subprocess.py::test_print[device_print-float16] -test/unit/language/test_subprocess.py::test_print[device_print-float32] -test/unit/language/test_subprocess.py::test_print[device_print-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float16] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float64] -test/unit/language/test_subprocess.py::test_print[device_print_scalar-float32] -# https://github.com/intel/intel-xpu-backend-for-triton/issues/1704 -test/unit/language/test_subprocess.py::test_print[device_print_uint-uint32]