diff --git a/.github/workflows/smoke-test.yml b/.github/workflows/smoke-test.yml index b81d1b2..24f2fb5 100644 --- a/.github/workflows/smoke-test.yml +++ b/.github/workflows/smoke-test.yml @@ -24,11 +24,17 @@ jobs: - name: Install package and dependencies run: uv sync --dev + - name: Clone FlagGems source + run: git clone https://github.com/FlagOpen/FlagGems.git + + - name: Build and install FlagGems + run: uv pip install FlagGems/ + - name: Clone FACTO source run: git clone https://github.com/pytorch-labs/FACTO.git - name: Build and install FACTO - run: cd FACTO && uv pip install . + run: uv pip install FACTO/ - name: Run smoke test run: uv run python -m BackendBench.scripts.main --suite smoke --backend aten diff --git a/BackendBench/backends/flag_gems.py b/BackendBench/backends/flag_gems.py index 91306af..ce40c1e 100644 --- a/BackendBench/backends/flag_gems.py +++ b/BackendBench/backends/flag_gems.py @@ -290,7 +290,7 @@ def __init__(self) -> None: torch.ops.aten.mse_loss.default: flag_gems.ops.mse_loss, torch.ops.aten.eye.default: flag_gems.ops.eye, torch.ops.aten.eye.m: flag_gems.ops.eye_m, - torch.ops.aten.to.dtype: flag_gems.ops.to_dtype, + # torch.ops.aten.to.dtype: flag_gems.ops.to_dtype, # skip as module 'flag_gems.ops' has no attribute 'to_dtype' } # Register all operators in the global registry to ensure consistent object identity for op in self.ops.keys():