Skip to content

GitLab

Commit 9edee0c0 authored by ljss's avatar ljss
Browse files

update .gitignore

parent 9c5dfab6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
.gitignore
View file @ 9edee0c0
...@@ -7,3 +7,4 @@ dist/ ...@@ -7,3 +7,4 @@ dist/
*.png *.png
/.vscode /.vscode
compile_commands.json compile_commands.json
.cache
setup.py
View file @ 9edee0c0
...@@ -11,10 +11,12 @@ from torch.utils.cpp_extension import ( ...@@ -11,10 +11,12 @@ from torch.utils.cpp_extension import (
IS_WINDOWS, IS_WINDOWS,
) )
def append_nvcc_threads(nvcc_extra_args): def append_nvcc_threads(nvcc_extra_args):
nvcc_threads = os.getenv("NVCC_THREADS") or "32" nvcc_threads = os.getenv("NVCC_THREADS") or "32"
return nvcc_extra_args + ["--threads", nvcc_threads] return nvcc_extra_args + ["--threads", nvcc_threads]
def get_features_args(): def get_features_args():
features_args = [] features_args = []
DISABLE_FP16 = os.getenv("FLASH_MLA_DISABLE_FP16", "FALSE") in ["TRUE", "1"] DISABLE_FP16 = os.getenv("FLASH_MLA_DISABLE_FP16", "FALSE") in ["TRUE", "1"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment