1
0
Fork 0
pull/1047/head
George Hotz 2023-06-25 15:24:26 -07:00
parent c8fbdeb48e
commit 0f281e7b18
2 changed files with 3 additions and 1 deletions

View File

@ -33,6 +33,8 @@ jobs:
python-version: 3.8
- name: Install dependencies
run: pip install -e '.[linting,testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Repo line count
run: python3 sz.py
- name: Lint with pylint
run: python -m pylint --disable=all -e W0311 --jobs=0 --indent-string=' ' **/*.py
- name: Lint with flake8

View File

@ -35,7 +35,7 @@ class TestLLaMASpeed(unittest.TestCase):
tms.append(time.perf_counter())
print(f"{st:15s} runtime in ms:", ', '.join("%.2f"%((tms[i+1]-tms[i])*1000) for i in range(len(tms)-1)))
run_llama("compile")
run_llama("codegen")
run_llama("methodcache", False)
pr = start_profile()