Skip to content

GitLab

Commit 2ed083ba authored by helloyongyang's avatar helloyongyang
Browse files

fix ci

parent c7bb59cd
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
lightx2v_kernel/python/lightx2v_kernel/__init__.py
View file @ 2ed083ba
import ctypes import ctypes
import os import os
import platform import platform
import torch import torch
from lightx2v_kernel import common_ops
from lightx2v_kernel.gemm import cutlass_scaled_fp4_mm, scaled_fp4_quant
from lightx2v_kernel.version import __version__
SYSTEM_ARCH = platform.machine() SYSTEM_ARCH = platform.machine()
...@@ -10,8 +13,5 @@ cuda_path = f"/usr/local/cuda/targets/{SYSTEM_ARCH}-linux/lib/libcudart.so.12" ...@@ -10,8 +13,5 @@ cuda_path = f"/usr/local/cuda/targets/{SYSTEM_ARCH}-linux/lib/libcudart.so.12"
if os.path.exists(cuda_path): if os.path.exists(cuda_path):
ctypes.CDLL(cuda_path, mode=ctypes.RTLD_GLOBAL) ctypes.CDLL(cuda_path, mode=ctypes.RTLD_GLOBAL)
from lightx2v_kernel import common_ops
from lightx2v_kernel.gemm import cutlass_scaled_fp4_mm, scaled_fp4_quant
from lightx2v_kernel.version import __version__
build_tree_kernel = None build_tree_kernel = None
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