Skip to content

Commit cb1e9ff

Browse files
committed
format python files
1 parent 6c0aa6f commit cb1e9ff

File tree

7 files changed

+5
-4
lines changed

7 files changed

+5
-4
lines changed

mlir/test/Examples/NVGPU/Ch0.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ def kernel():
4242
# 3. Call the GPU kernel
4343
kernel()
4444

45+
4546
alpha = 100
4647
# 4. The `mlir_func` decorator JIT compiles the IR and executes the MLIR function.
4748
main(alpha)

mlir/test/Examples/NVGPU/Ch1.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
import numpy as np
2525

2626

27-
2827
@NVDSL.mlir_func
2928
def saxpy(x, y, alpha):
3029
# 1. Use MLIR GPU dialect to allocate and copy memory

mlir/test/Examples/NVGPU/Ch2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from mlir.extras import types as T
2929
import numpy as np
3030

31+
3132
@NVDSL.mlir_func
3233
def saxpy(x, y, alpha):
3334
token_ty = gpu.AsyncTokenType.get()

mlir/test/Examples/NVGPU/Ch3.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from mlir.extras import types as T
2626
import numpy as np
2727

28+
2829
def tma_load(
2930
mbar_group: Mbarriers,
3031
a_tma: TMA,

mlir/test/Examples/NVGPU/Ch4.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@
5252
import numpy as np
5353

5454

55-
5655
def partition_shape():
5756
"""
5857
Calculate the partition shape based on the block IDs.

mlir/test/Examples/NVGPU/Ch5.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@
5252
import numpy as np
5353

5454

55-
5655
def partition_shape():
5756
"""
5857
Calculate the partition shape based on the block IDs.
@@ -479,4 +478,4 @@ def gemm_warp_specialized_kernel():
479478
# DUMPIR: memref.store %{{.*}}, %[[SUBVIEW]][%arg15, %[[TID_EPI]]] : memref<128x128xf32, strided<[256, 1], offset: ?>>
480479
# DUMPIR: }
481480
# DUMPIR: }
482-
# DUMPIR: gpu.terminator
481+
# DUMPIR: gpu.terminator

mlir/test/Examples/NVGPU/tools/nvdsl.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
MLIR_DYNAMIC = -9223372036854775808
1212
DUMP_ONLY = os.getenv("MLIR_NVDSL_PRINT_IR") == "1"
1313

14+
1415
def const(value: int, ty=None):
1516
ty = T.index() if ty is None else ty
1617
if isinstance(value, ir.Value) and (

0 commit comments

Comments
 (0)