setup.py

import os
import re

from setuptools import setup

PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))

def read(*path):
    return open(os.path.join(*path), encoding="utf8").read()

def find_version(filepath: str) -> str:
    """Extract version information from the given filepath.

    Adapted from https://github.com/ray-project/ray/blob/0b190ee1160eeca9796bc091e07eaebf4c85b511/python/setup.py
    """
    with open(filepath) as fp:
        version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]",
                                  fp.read(), re.M)
        if version_match:
            return version_match.group(1)
        raise RuntimeError("Unable to find version string.")

setup(
    name="block_math",
    version=find_version(os.path.join(PROJECT_ROOT, "block_math", "__init__.py")),
    author="Myeonghwan Ahn",
    description="Architecture-agnostic microscaling(MX)-like math in Triton-lang",
    long_description=read(PROJECT_ROOT, "README.md"),
    long_description_content_type="text/markdown",
    packages=["block_math"]
)