Skip to content

PyTorch integration test at threshold=1 on aarch64 #6

Description

@SonicField

PyTorch integration test at threshold=1

Phase 5f enables generator JIT compilation on aarch64 (30/33 tests pass). The terminal goal is PyTorch performance on ARM. The synthetic test suite proves generator mechanics work; a PyTorch integration test proves the target workload works.

Proposed test

Run a PyTorch training workload on devgpu004 with:

PYTHONJITCOMPILATIONTHRESHOLD=1 python train.py

Success criteria

  • No crashes or incorrect results
  • Generators in dataloaders and training loops are JIT-compiled
  • Performance is at least as good as without generator JIT (no regressions)

Why this matters

This validates that the aarch64 generator JIT fix works in the target application, not just synthetic tests.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions