Skip to content

Commit 3ce0b81

Browse files
committed
Rename package name to native_sparse_attention
1 parent c776771 commit 3ce0b81

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

pyproject.toml

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[project]
2-
name = "fla"
2+
name = "native_sparse_attention"
33
dynamic = ["version"]
44
description = "Trtion kernels for Native Sparse Attention"
55
readme = "README.md"
@@ -29,7 +29,7 @@ conv1d = ["causal-conv1d>=1.4.0"]
2929
dev = ["pytest"]
3030

3131
[project.urls]
32-
Homepage = "https://github.com/fla-org/flash-linear-attention"
32+
Homepage = "https://github.com/fla-org/native-sparse-attention"
3333

3434
[build-system]
3535
requires = ["setuptools>=45", "wheel", "ninja", "torch"]

setup.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -12,20 +12,20 @@
1212

1313

1414
def get_package_version():
15-
with open(Path(os.path.dirname(os.path.abspath(__file__))) / 'fla' / '__init__.py') as f:
15+
with open(Path(os.path.dirname(os.path.abspath(__file__))) / 'native_sparse_attention' / '__init__.py') as f:
1616
version_match = re.search(r"^__version__\s*=\s*(.*)$", f.read(), re.MULTILINE)
1717
return ast.literal_eval(version_match.group(1))
1818

1919

2020
setup(
21-
name='fla',
21+
name='native_sparse_attention',
2222
version=get_package_version(),
2323
description='Trtion kernels for Native Sparse Attention',
2424
long_description=long_description,
2525
long_description_content_type='text/markdown',
2626
author='Songlin Yang, Yu Zhang',
2727
28-
url='https://github.com/fla-org/flash-linear-attention',
28+
url='https://github.com/fla-org/native-sparse-attention',
2929
packages=find_packages(),
3030
license='MIT',
3131
classifiers=[

0 commit comments

Comments
 (0)