unsloth/.github
Daniel Han 888355b24a CI(consolidated): route bare unsloth_zoo imports through pytest shim files
Run #6 progressed past install / collection but failed at step 10
(test_apply_fused_lm_head) inside unsloth_zoo/temporary_patches/gpt_oss.py:1141:

    device_memory = torch.cuda.memory.mem_get_info(0)[-1]
    AssertionError: Torch not compiled with CUDA enabled

The bare `python -c` heredoc spoofed torch.cuda.is_available but not the
deeper torch.cuda.memory.mem_get_info / cudart() lazy_init path. The
existing tests/conftest.py:84-141 already has the full spoof.

Switching three steps to write a one-shot shim test file under tests/ and
run it via pytest — pytest walks UP and applies tests/conftest.py before
the unsloth_zoo.* import, so the full GPU-spoof harness covers the deeper
mem_get_info / get_device_capability / is_bf16_supported probes:

- Step "test_apply_fused_lm_head": tests/_zoo_apply_fused_lm_head_shim.py
- Step "Runtime checks — invoke every zero-arg patch_*": tests/_runtime_patch_check_shim.py
- Step "Runtime checks — patch_tiled_mlp on a synthetic MLP module":
  tests/_tiled_mlp_check_shim.py

Each shim is rm-ed at the end of its step so it never lands in a commit.

Locally re-validated test_apply_fused_lm_head shim: 1 passed in 3.47 s.
2026-05-07 05:07:59 +00:00
..
ISSUE_TEMPLATE Update issue template 2026-03-23 10:10:15 +05:30
workflows CI(consolidated): route bare unsloth_zoo imports through pytest shim files 2026-05-07 05:07:59 +00:00
CODEOWNERS Update CODEOWNERS 2026-03-13 13:38:19 -07:00
dependabot.yml CI(security): defense-in-depth additions across 7 axes 2026-05-07 00:09:40 +00:00
FUNDING.yml Update FUNDING.yml (#3792) 2025-12-28 19:57:43 -08:00