From eb812c205b4a4327230f5e75407d06e75917417b Mon Sep 17 00:00:00 2001
From: Pierce Freeman <piercefreeman@gmail.com>
Date: Wed, 7 Jun 2023 17:20:13 -0700
Subject: [PATCH] Remove builder project

---
 flash_attn_builder/README.md                  |  3 --
 .../flash_attn_builder/__init__.py            |  0
 flash_attn_builder/flash_attn_builder/main.py | 54 -------------------
 flash_attn_builder/pyproject.toml             | 15 ------
 4 files changed, 72 deletions(-)
 delete mode 100644 flash_attn_builder/README.md
 delete mode 100644 flash_attn_builder/flash_attn_builder/__init__.py
 delete mode 100644 flash_attn_builder/flash_attn_builder/main.py
 delete mode 100644 flash_attn_builder/pyproject.toml

diff --git a/flash_attn_builder/README.md b/flash_attn_builder/README.md
deleted file mode 100644
index 3e42b3b..0000000
--- a/flash_attn_builder/README.md
+++ /dev/null
@@ -1,3 +0,0 @@
-## flash-attn-builder
-
-Basic build utilities for flash-attn.
diff --git a/flash_attn_builder/flash_attn_builder/__init__.py b/flash_attn_builder/flash_attn_builder/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/flash_attn_builder/flash_attn_builder/main.py b/flash_attn_builder/flash_attn_builder/main.py
deleted file mode 100644
index 1e750e7..0000000
--- a/flash_attn_builder/flash_attn_builder/main.py
+++ /dev/null
@@ -1,54 +0,0 @@
-import os
-import sys
-import urllib
-import setuptools.build_meta
-from setuptools.command.install import install
-from packaging.version import parse, Version
-
-# @pierce - TODO: Update for proper release
-BASE_WHEEL_URL = "https://github.com/piercefreeman/flash-attention/releases/download/{tag_name}/{wheel_name}"
-
-# FORCE_BUILD: Force a fresh build locally, instead of attempting to find prebuilt wheels
-# SKIP_CUDA_BUILD: Intended to allow CI to use a simple `python setup.py sdist` run to copy over raw files, without any cuda compilation
-FORCE_BUILD = os.getenv("FLASH_ATTENTION_FORCE_BUILD", "FALSE") == "TRUE"
-
-class CustomBuildBackend(setuptools.build_meta._BuildMetaBackend):
-
-    def build_wheel(self, wheel_directory, config_settings=None, metadata_directory=None):
-        this_file_directory = os.path.dirname(os.path.abspath(__file__))
-        print(f'This file is located in: {this_file_directory}')
-
-        sys.argv = [
-            *sys.argv[:1],
-            *self._global_args(config_settings),
-            *self._arbitrary_args(config_settings),
-        ]
-        with setuptools.build_meta.no_install_setup_requires():
-            self.run_setup()
-
-        print("OS", os.environ["FLASH_ATTENTION_WHEEL_URL"])
-        print("config_settings", config_settings)
-        print("metadata_directory", metadata_directory)
-        raise ValueError
-
-        print("Guessing wheel URL: ", wheel_url)
-        
-        try:
-            urllib.request.urlretrieve(wheel_url, wheel_filename)
-            os.system(f'pip install {wheel_filename}')
-            os.remove(wheel_filename)
-        except urllib.error.HTTPError:
-            print("Precompiled wheel not found. Building from source...")
-            # If the wheel could not be downloaded, build from source
-            super().build_wheel(wheel_directory, config_settings, metadata_directory)
-
-
-_BACKEND = CustomBuildBackend()  # noqa
-
-
-get_requires_for_build_wheel = _BACKEND.get_requires_for_build_wheel
-get_requires_for_build_sdist = _BACKEND.get_requires_for_build_sdist
-prepare_metadata_for_build_wheel = _BACKEND.prepare_metadata_for_build_wheel
-build_wheel = _BACKEND.build_wheel
-build_sdist = _BACKEND.build_sdist
-
diff --git a/flash_attn_builder/pyproject.toml b/flash_attn_builder/pyproject.toml
deleted file mode 100644
index 7fa99d4..0000000
--- a/flash_attn_builder/pyproject.toml
+++ /dev/null
@@ -1,15 +0,0 @@
-[tool.poetry]
-name = "flash-attn-builder"
-version = "0.1.0"
-description = ""
-authors = ["Pierce Freeman <pierce@freeman.vc>"]
-readme = "README.md"
-packages = [{include = "flash_attn_builder"}]
-
-[tool.poetry.dependencies]
-python = "^3.10"
-
-
-[build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"