[FRONTEND] Added on-disk cache for compiled kernels (#287)

2021-09-18 22:48:26 -07:00
parent bd855ac13d
commit 6e5b0b4301
5 changed files with 235 additions and 81 deletions
--- a/python/setup.py
+++ b/python/setup.py
@@ -127,7 +127,7 @@ 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=["torch"],
+    install_requires=["torch", "filelock"],
    package_data={"triton/ops": ["*.c"], "triton/ops/blocksparse": ["*.c"]},
    include_package_data=True,
    ext_modules=[CMakeExtension("triton", "triton/_C/")],