mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
df08301e76
I've add an option to yapf to do what we want for long lines, see https://github.com/google/yapf/pull/1177. We can now have a real Python formatter, yay! To make this PR, I ran my modified yapf over the repository, then looked over the full diff. Where yapf was mangling the param list of long function decls/calls (mostly kernels), I manually added `#` to put linebreaks where we want. I fixed up other formatting too -- mostly adding or removing a trailing comma from lists. Overall, trailing `#` was sufficient to get formatting similar to our current code. I didn't have to disable yapf anywhere. --------- Co-authored-by: Phil Tillet <phil@openai.com>
19 lines
440 B
Python
19 lines
440 B
Python
from .autotuner import (Autotuner, Config, Heuristics, OutOfResources, autotune, heuristics)
|
|
from .driver import driver
|
|
from .jit import JITFunction, KernelInterface, MockTensor, TensorWrapper, reinterpret
|
|
|
|
__all__ = [
|
|
"driver",
|
|
"Config",
|
|
"Heuristics",
|
|
"autotune",
|
|
"heuristics",
|
|
"JITFunction",
|
|
"KernelInterface",
|
|
"reinterpret",
|
|
"TensorWrapper",
|
|
"OutOfResources",
|
|
"MockTensor",
|
|
"Autotuner",
|
|
]
|