Unverified Commit 81c53ef5 authored by Ye (Charlotte) Qi's avatar Ye (Charlotte) Qi Committed by GitHub
Browse files

[Misc] collect flashinfer version in collect_env.py (#24378)


Signed-off-by: default avatarYe (Charlotte) Qi <yeq@meta.com>
parent 75334956
...@@ -74,6 +74,7 @@ DEFAULT_CONDA_PATTERNS = { ...@@ -74,6 +74,7 @@ DEFAULT_CONDA_PATTERNS = {
"zmq", "zmq",
"nvidia", "nvidia",
"pynvml", "pynvml",
"flashinfer-python",
} }
DEFAULT_PIP_PATTERNS = { DEFAULT_PIP_PATTERNS = {
...@@ -89,6 +90,7 @@ DEFAULT_PIP_PATTERNS = { ...@@ -89,6 +90,7 @@ DEFAULT_PIP_PATTERNS = {
"zmq", "zmq",
"nvidia", "nvidia",
"pynvml", "pynvml",
"flashinfer-python",
} }
......
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