mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
0fd59d64ab
* add fmha repo Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix format Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix code style Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix header Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix header kernel_traits.h Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * add .gitignore file Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * add SLIDING_WINDOW_ATTENTION Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix style Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * fix format Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * update setup.py Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> * update build_wheel.py Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> --------- Signed-off-by: Qidi Sang <200703406+qsang-nv@users.noreply.github.com> Signed-off-by: qsang-nv <200703406+qsang-nv@users.noreply.github.com>
12 lines
405 B
Bash
Executable File
12 lines
405 B
Bash
Executable File
#export CUDAFLAGS=-I/data/sw/cublas_28254210/cublas/src/
|
|
#export CXXFLAGS="-L/data/sw/cublas_28254210/bin/x86_64_Linux_release $CUDAFLAGS"
|
|
|
|
#export CUDA=/data/sw/cuda_11_0_170_internal/
|
|
#export PATH=/data/sw/cuda_11_0_170_internal/bin:$PATH
|
|
|
|
export CUDA=/data/sw/cuda_11_1_52/
|
|
export PATH=/data/sw/cuda_11_1_52/bin:$PATH
|
|
|
|
#export CUDA=/data/sw/cuda_11_0_228/
|
|
#export PATH=/data/sw/cuda_11_0_228/bin:$PATH
|