Unverified Commit c37c0af9 authored by Jiangyun Zhu's avatar Jiangyun Zhu Committed by GitHub
Browse files

[Misc] Fix comments in `tests/kernels/quantization` (#23675)


Signed-off-by: default avatarzjy0516 <riverclouds.zhu@qq.com>
parent 9715f7bb
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for the AWQ Triton kernel. """Tests for the AWQ Triton kernel.
Run `pytest tests/kernels/test_awq_triton.py`. Run `pytest tests/kernels/quantization/test_awq_triton.py`.
""" """
import pytest import pytest
import torch import torch
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for sparse cutlass kernels """Tests for sparse cutlass kernels
Run `pytest tests/kernels/test_semi_structured.py`. Run `pytest tests/kernels/quantization/test_cutlass_2of4_sparse.py`.
""" """
import pytest import pytest
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for cutlass kernels """Tests for cutlass kernels
Run `pytest tests/kernels/test_cutlass.py`. Run `pytest tests/kernels/quantization/test_cutlass_scaled_mm.py`.
""" """
import random import random
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for the CUTLASS W4A8 kernel. """Tests for the CUTLASS W4A8 kernel.
Run `pytest tests/kernels/test_cutlass_w4a8.py`. Run `pytest tests/kernels/quantization/test_cutlass_w4a8.py`.
""" """
from dataclasses import dataclass from dataclasses import dataclass
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for the machete kernel. """Tests for the machete kernel.
Run `pytest tests/kernels/test_machete_mm.py`. Run `pytest tests/kernels/quantization/test_machete_mm.py`.
""" """
import math import math
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for the marlin kernel. """Tests for the marlin kernel.
Run `pytest tests/kernels/marlin/test_marlin_gemm.py`. Run `pytest tests/kernels/quantization/test_marlin_gemm.py`.
""" """
import pytest import pytest
import torch import torch
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Tests for the triton_scaled_mm kernel """Tests for the triton_scaled_mm kernel
Run `pytest tests/kernels/test_triton_scaled_mm.py`. Run `pytest tests/kernels/quantization/test_triton_scaled_mm.py`.
""" """
import importlib import importlib
from typing import Optional from typing import Optional
......
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