File size: 2,299 Bytes
9b43109 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | """The shipped artifacts are real binaries.
A `.so` committed while its working copy was an unsmudged LFS pointer stores the pointer text as the
object: a ~130-byte file that fails to `dlopen`. It has happened once, silently, and it is invisible
to every other test because loading such a build looks like "no build for this platform".
"""
from pathlib import Path
import pytest
BUILD = Path(__file__).resolve().parent.parent / "build"
ARTIFACTS = sorted(BUILD.glob("*/*.so")) if BUILD.is_dir() else []
@pytest.mark.skipif(not ARTIFACTS, reason="no built artifacts in this checkout")
@pytest.mark.parametrize("so", ARTIFACTS, ids=lambda p: p.parent.name)
def test_artifact_is_a_binary_not_an_lfs_pointer(so):
head = so.read_bytes()[:64]
assert not head.startswith(b"version https://git-lfs"), (
f"{so.relative_to(BUILD)} is an LFS pointer ({so.stat().st_size} bytes), not a library. "
"It was committed while unsmudged; restore it from the last commit whose pointer records a "
"real size, and re-commit."
)
assert so.stat().st_size > 100_000, f"{so.relative_to(BUILD)} is only {so.stat().st_size} bytes"
@pytest.mark.skipif(not ARTIFACTS, reason="no built artifacts in this checkout")
def test_the_build_for_this_platform_loads():
"""A build that ships for this machine must import; "will not load" is not "not built".
Without this, a corrupt artifact makes every other test skip — the suite reports green for a
package that cannot be used at all.
"""
import platform
import torch
from kernels import get_local_kernel
arch = "aarch64" if platform.machine() in ("arm64", "aarch64") else "x86_64"
system = "darwin" if platform.system() == "Darwin" else "linux"
backend = "metal" if system == "darwin" else "cuda"
major, minor = torch.__version__.split(".")[:2]
match = [
d for d in sorted(BUILD.iterdir())
if d.name.startswith(f"torch{major}{minor}-") and backend in d.name and arch in d.name
]
if not match:
pytest.skip(f"no torch{major}{minor} {backend} {arch} build in this checkout")
try:
get_local_kernel(BUILD.parent, backend)
except Exception as error: # noqa: BLE001
pytest.fail(f"{match[0].name} ships but will not load: {error}")
|