Unverified Commit 7a0b0985 authored by Yang Yong (雍洋)'s avatar Yang Yong (雍洋) Committed by GitHub
Browse files

Update lightx2v_kernel (#458)

parent 82941fcc
...@@ -26,16 +26,7 @@ if(CUTLASS_PATH) ...@@ -26,16 +26,7 @@ if(CUTLASS_PATH)
set(repo-cutlass_SOURCE_DIR ${CUTLASS_PATH}) set(repo-cutlass_SOURCE_DIR ${CUTLASS_PATH})
message(STATUS "Using local CUTLASS from: ${CUTLASS_PATH}") message(STATUS "Using local CUTLASS from: ${CUTLASS_PATH}")
else() else()
message(STATUS "Start to git clone CUTLASS from GitHub...") message(FATAL_ERROR "CUTLASS_PATH is not set. Please manually download CUTLASS first.")
include(FetchContent)
FetchContent_Declare(
repo-cutlass
GIT_REPOSITORY https://github.com/NVIDIA/cutlass
GIT_TAG b995f933179c22d3fe0d871c3a53d11e4681950f
GIT_SHALLOW OFF
)
FetchContent_MakeAvailable(repo-cutlass)
message(STATUS "Using CUTLASS from ${repo-cutlass_SOURCE_DIR}")
endif() endif()
......
...@@ -8,17 +8,16 @@ pip install scikit_build_core uv ...@@ -8,17 +8,16 @@ pip install scikit_build_core uv
``` ```
### Build whl ### Build whl
```
MAX_JOBS=$(nproc) && CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
uv build --wheel \
-Cbuild-dir=build . \
--verbose \
--color=always \
--no-build-isolation
```
During the above build process, the cutlass source code will be downloaded automatically. If you have already downloaded the source code, you can specify the local cutlass path:
``` ```
git clone https://github.com/NVIDIA/cutlass.git
git clone https://github.com/ModelTC/LightX2V.git
cd LightX2V/lightx2v_kernel
# Set the /path/to/cutlass below to the absolute path of cutlass you download.
MAX_JOBS=$(nproc) && CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \ MAX_JOBS=$(nproc) && CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
uv build --wheel \ uv build --wheel \
-Cbuild-dir=build . \ -Cbuild-dir=build . \
......
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