apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-30 03:22:05 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

polish code with new pre-commit rule (#2923)

Browse Source
This commit is contained in:
Zero Rains
2025-07-19 23:19:27 +08:00
committed by GitHub
parent b8676d71a8
commit 25698d56d1
424 changed files with 14307 additions and 13518 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/tune_scaled_gemm_f8_i4_f16.py
View File

@@ -14,14 +14,14 @@
"""tune_cutlass_fp8int4_gemm"""
import os
import paddle
from fastdeploy.model_executor.ops.gpu import scaled_gemm_f8_i4_f16
from tqdm import tqdm
from fastdeploy.model_executor.ops.gpu import scaled_gemm_f8_i4_f16
def tune_scaled_gemm_f8_i4_f16(
ns: list, ks: list, dtype="int8", is_test=True, is_read_from_file=False
):
def tune_scaled_gemm_f8_i4_f16(ns: list, ks: list, dtype="int8", is_test=True, is_read_from_file=False):
"""
Tune fp8 int4 gemm.
"""