diff --git a/ffi/pyproject.toml b/ffi/pyproject.toml index 430b47c33b8b..79cd95878666 100644 --- a/ffi/pyproject.toml +++ b/ffi/pyproject.toml @@ -42,8 +42,9 @@ GitHub = "https://github.com/apache/tvm/ffi" [project.optional-dependencies] # setup tools is needed by torch jit for best perf -torch = ["torch", "setuptools"] -test = ["pytest", "numpy", "torch"] +torch = ["torch", "setuptools", "ninja"] +cpp = ["ninja"] +test = ["pytest", "numpy", "torch", "ninja"] [project.scripts] tvm-ffi-config = "tvm_ffi.config:__main__" diff --git a/ffi/tests/python/test_load_inline.py b/ffi/tests/python/test_load_inline.py index f809cede5927..c35ebd30e225 100644 --- a/ffi/tests/python/test_load_inline.py +++ b/ffi/tests/python/test_load_inline.py @@ -17,6 +17,7 @@ import pytest import numpy +import sys try: import torch @@ -27,6 +28,7 @@ from tvm_ffi.module import Module +@pytest.mark.xfail(not sys.platform.startswith("linux"), reason="need to support non-linux") def test_load_inline_cpp(): mod: Module = tvm_ffi.cpp.load_inline( name="hello", @@ -53,6 +55,7 @@ def test_load_inline_cpp(): numpy.testing.assert_equal(x + 1, y) +@pytest.mark.xfail(not sys.platform.startswith("linux"), reason="need to support non-linux") def test_load_inline_cpp_with_docstrings(): mod: Module = tvm_ffi.cpp.load_inline( name="hello", @@ -79,6 +82,7 @@ def test_load_inline_cpp_with_docstrings(): numpy.testing.assert_equal(x + 1, y) +@pytest.mark.xfail(not sys.platform.startswith("linux"), reason="need to support non-linux") def test_load_inline_cpp_multiple_sources(): mod: Module = tvm_ffi.cpp.load_inline( name="hello", @@ -121,6 +125,7 @@ def test_load_inline_cpp_multiple_sources(): numpy.testing.assert_equal(x + 1, y) +@pytest.mark.xfail(not sys.platform.startswith("linux"), reason="need to support non-linux") def test_load_inline_cpp_build_dir(): mod: Module = tvm_ffi.cpp.load_inline( name="hello",