Skip to content

Prerelease: v1.1.1

Pre-release
Pre-release

Choose a tag to compare

View all tags
@asgersvenning asgersvenning released this 28 Apr 17:50
· 35 commits to develop since this release
v1.1-beta.1
18ef657

Overview

Fixed broken CLI API and improved JIT function @torch.compile(...) decorator usage. Also greatly improved README installation guide for CUDA compatible installlation.

Changes

  • Explictly run GitHub Action for unit tests with CPU.
  • Moved CLI scripts from /src/bin/**<SCRIPT>** to /src/flat_bug/cli/<SCRIPT> submodule. (and change pyproject.toml to use the new path)
  • Improved README installation guide, particularly with steps related to PyTorch integration with uv/pip. (https://docs.astral.sh/uv/guides/integration/pytorch/#configuring-accelerators-with-optional-dependencies)
  • Change all use of @torch.compile to @torch.compile(fullgraph=True, mode="max-autotune-no-cudagraphs").
  • Add torch.set_float32_matmul_precision("medium") to the start of flat_bug.cli.fb_predict.py:main i.e. the main CLI API for this package.

Full Changelog: v1.1-beta.0...v1.1-beta.1

Issues? See #151

Assets 2
Loading