[DOCS] fix tutorials for v2.0 (#422)
- Fix meta-parameter usage on tutorials. - Install tutorial dependencies on CI. - Switch from `requirements-test.txt` to `extras_require` for test dependencies, and also use it for tutorial dependencies. - Make some performance tests deterministic.
This commit is contained in:
committed by
GitHub
parent
8bf551ae7a
commit
9801aa7b56
@@ -126,7 +126,11 @@ setup(
|
||||
description="A language and compiler for custom Deep Learning operations",
|
||||
long_description="",
|
||||
packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/ops", "triton/ops/blocksparse"],
|
||||
install_requires=["cmake", "torch", "filelock"],
|
||||
install_requires=[
|
||||
"cmake",
|
||||
"filelock",
|
||||
"torch",
|
||||
],
|
||||
package_data={"triton/ops": ["*.c"], "triton/ops/blocksparse": ["*.c"]},
|
||||
include_package_data=True,
|
||||
ext_modules=[CMakeExtension("triton", "triton/_C/")],
|
||||
@@ -142,4 +146,16 @@ setup(
|
||||
"License :: OSI Approved :: MIT License",
|
||||
"Programming Language :: Python :: 3.6",
|
||||
],
|
||||
extras_require={
|
||||
"tests": [
|
||||
"numpy",
|
||||
"pytest",
|
||||
"scipy>=1.7.1",
|
||||
],
|
||||
"tutorials": [
|
||||
"matplotlib",
|
||||
"pandas",
|
||||
"tabulate",
|
||||
],
|
||||
},
|
||||
)
|
||||
|
Reference in New Issue
Block a user