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/
*.png
/.vscode
compile_commands.json
.cache
setup.py
View file @ 9edee0c0
......@@ -11,10 +11,12 @@ from torch.utils.cpp_extension import (
IS_WINDOWS,
)
def append_nvcc_threads(nvcc_extra_args):
nvcc_threads = os.getenv("NVCC_THREADS") or "32"
return nvcc_extra_args + ["--threads", nvcc_threads]
def get_features_args():
features_args = []
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