nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[kernel] Reorg and fix fastvideo-kernel (#962)

Browse Source 
Co-authored-by: ShreejithSG <shreejithsg@gmail.com>
This commit is contained in:
William Lin
2026-01-22 12:06:40 +00:00
committed by RandNMR73
co-authored by ShreejithSG
parent 8f4cd10b49
commit 254b49cdb7
72 changed files with 1068 additions and 6633 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.python3.12
+3 -10
View File
@@ -55,18 +55,11 @@ RUN source $HOME/.local/bin/env && \
echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \
echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile
# Install STA (Sliding Tile Attention)
# Install FastVideo Unified Kernel
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn/sliding_tile_attn && \
cd fastvideo-kernel && \
git submodule update --init --recursive && \
python setup.py install
# Install VSA
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn/video_sparse_attn && \
git submodule update --init --recursive && \
python setup.py install
./build.sh
EXPOSE 22