UV Working Install

[stevenwalton]

Converted install process to work with uv and virtual environments. While doing so I performed general upgrades to the setup file and our project file will make NATTEN more easily searchable on pypi

Changes

• Created the pyproject.toml file
  • Also introduced classifiers for pypi. Compare torch to natten on left hand side.
• Reformatted setup.py: This file should now be easier to edit for future architectures and flexible to other types of changes. Please review
• Makefile now loads programs from the user's environment. This was the main issue previously as it would build natten into the wrong python version if you used environments.
• CMakeLists was missing the ability to pass CMAKE_CUDA_FLAGS from the setup.py file. Now we can. Do we can run CMAKE_CUDA_FLAGS="-Wno-deprecated-declarations" uv pip install -e . if we want to turn off the cutlass deprecation warnings while building.
• Removed requirements.txt: now handled by pyproject.toml
• Removed requirements-dev.txt: now handled by pyproject.toml
• Removed setup.cfg: No longer needed
• Added uv.lock
• Documented setup.py

Build process

$ git clone --recursive https://github.com/SHI-Labs/NATTEN
$ cd NATTEN

Option 1: Explicit

# Create the venv (default path is .venv)
# --seed means we pull in pip and other tools
# -p == --python
$ uv venv --seed -p 3.9

# Load venv
$ source .venv/bin/activate

# Install natten. Will do dependency resolution
# uv isn't needed because we did --seed
$ uv pip install -e .

# Get some tea, it'll be a bit

Option 2: One liner

# Create a venv based on pyproject.toml 
# We pass python version as by default it will use your most current version
$ uv sync -p 3.9

# Get some tea, it is installing natten too

Option 3: Build

# Do Option 1 but instead of the pip line 
$ uv build

# Get some tea

# Install wheel
$  uv pip install dist/natten-0.21.1-cp39-cp39-linux_x86_64.whl

After each of these options you may run the profile commands

Verified systems

• OSX M2 Air
• EndeavourOS with 4080S (sm89)

TODO:

• @alihassanijr , you will need to edit pyproject.toml on lines 22 and 25 to add your email address. We can either keep authors as is or include all contributors. Maintainers is what pypi shows and should stay you.
• Review
• Test on other architectures (should be fine)
  • SM90
  • SM100
• Clean up as we may not need some other files and dev dependencies like flake