Unverified Commit 5e529522 authored by Yichen Yan's avatar Yichen Yan Committed by GitHub
Browse files

[Lint] Add ruff config to check for useless spaces (#807)

* update lint config

* Remove spaces for blank line

* update
parent 4d54854b
...@@ -43,7 +43,7 @@ def requires_cuda_compute_version(major_version, minor_version=0, mode="ge"): ...@@ -43,7 +43,7 @@ def requires_cuda_compute_version(major_version, minor_version=0, mode="ge"):
minor_version: int minor_version: int
The minor version of the (major,minor) version tuple. The minor version of the (major,minor) version tuple.
mode: str mode: str
The mode of the comparison. The mode of the comparison.
......
...@@ -70,7 +70,7 @@ def InjectSoftwarePipeline(): ...@@ -70,7 +70,7 @@ def InjectSoftwarePipeline():
def InjectAssumes(): def InjectAssumes():
"""Inject Assumes """Inject Assumes
Returns: Returns:
------- -------
fpass : tvm.transform.Pass fpass : tvm.transform.Pass
...@@ -418,10 +418,10 @@ def LowerThreadAllreduce(): ...@@ -418,10 +418,10 @@ def LowerThreadAllreduce():
def LowerDeviceKernelLaunch(): def LowerDeviceKernelLaunch():
""" """
Create and return a transform pass that lowers device kernel launch constructs to target-specific IR. Create and return a transform pass that lowers device kernel launch constructs to target-specific IR.
This pass transforms high-level device kernel launch and related intrinsics into lower-level This pass transforms high-level device kernel launch and related intrinsics into lower-level
IR suitable for backend code generation and device-side lowering. IR suitable for backend code generation and device-side lowering.
Returns: Returns:
tvm.transform.Pass: The transform pass that performs device kernel launch lowering. tvm.transform.Pass: The transform pass that performs device kernel launch lowering.
""" """
...@@ -431,9 +431,9 @@ def LowerDeviceKernelLaunch(): ...@@ -431,9 +431,9 @@ def LowerDeviceKernelLaunch():
def LayoutReducer(): def LayoutReducer():
""" """
Return a TVM transform pass that performs layout reduction/normalization. Return a TVM transform pass that performs layout reduction/normalization.
This wrapper delegates to the underlying FFI implementation and returns a pass object suitable for use in a PassContext or pass pipeline. The pass is intended to simplify or reduce tensor/layout-related representations during relay/tile transformations. This wrapper delegates to the underlying FFI implementation and returns a pass object suitable for use in a PassContext or pass pipeline. The pass is intended to simplify or reduce tensor/layout-related representations during relay/tile transformations.
Returns: Returns:
The transform pass object produced by the FFI backend. The transform pass object produced by the FFI backend.
""" """
......
...@@ -22,7 +22,7 @@ class PassConfigKey(str, Enum): ...@@ -22,7 +22,7 @@ class PassConfigKey(str, Enum):
"""Disable fast math optimization. Default: False""" """Disable fast math optimization. Default: False"""
TL_PTXAS_REGISTER_USAGE_LEVEL = "tl.ptxas_register_usage_level" TL_PTXAS_REGISTER_USAGE_LEVEL = "tl.ptxas_register_usage_level"
"""The PTXAS register usage level in [0, 10], which controls the """The PTXAS register usage level in [0, 10], which controls the
aggressiveness of optimizations that affect register usage. Default: None""" aggressiveness of optimizations that affect register usage. Default: None"""
TL_ENABLE_PTXAS_VERBOSE_OUTPUT = "tl.enable_ptxas_verbose_output" TL_ENABLE_PTXAS_VERBOSE_OUTPUT = "tl.enable_ptxas_verbose_output"
......
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