Skip to content

Commit 9cb7250

Browse files
committed
[mlir][sparse][gpu] fixed some whitespace
Reviewed By: K-Wu Differential Revision: https://reviews.llvm.org/D157448
1 parent 475938d commit 9cb7250

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

mlir/include/mlir/Dialect/GPU/IR/GPUOps.td

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1776,7 +1776,7 @@ def GPU_Create2To4SpMatOp : GPU_Op<"create_2to4_spmat", [GPU_AsyncOpInterface]>
17761776
Index:$cols,
17771777
GPU_Prune2To4SpMatFlagAttr:$pruneFlag,
17781778
AnyMemRef:$memref);
1779-
let results = (outs Res<GPU_SparseSpMatHandle>:$spMat,
1779+
let results = (outs Res<GPU_SparseSpMatHandle>:$spMat,
17801780
Optional<GPU_AsyncToken>:$asyncToken);
17811781

17821782
let assemblyFormat = [{
@@ -2176,14 +2176,14 @@ def GPU_SpGEMMWorkEstimationOrComputeKind : I32EnumAttr<"SpGEMMWorkEstimationOrC
21762176
let cppNamespace = GPU_Dialect.cppNamespace;
21772177
}
21782178

2179-
def GPU_SpGEMMWorkEstimationOrComputeKindAttr : EnumAttr<GPU_Dialect,
2180-
GPU_SpGEMMWorkEstimationOrComputeKind,
2179+
def GPU_SpGEMMWorkEstimationOrComputeKindAttr : EnumAttr<GPU_Dialect,
2180+
GPU_SpGEMMWorkEstimationOrComputeKind,
21812181
"spgemm_work_estimation_or_compute_kind"> {}
21822182

21832183
def GPU_SpGEMMCreateDescrOp : GPU_Op<"spgemm_create_descr", [GPU_AsyncOpInterface]> {
21842184
let summary = "SpGEMM Create Descr operation";
21852185
let description = [{
2186-
The `gpu.spgemm_create_descr` creates a descriptor for the SpGEMM operation.
2186+
The `gpu.spgemm_create_descr` creates a descriptor for the SpGEMM operation.
21872187
The descriptor describes the SpGEMM operation and stores the internal data
21882188
throughout the computation. It needs to be passed as an argument to
21892189
spgemm_* operations.
@@ -2238,11 +2238,11 @@ def GPU_SpGEMMDestroyDescrOp : GPU_Op<"spgemm_destroy_descr", [GPU_AsyncOpInterf
22382238
def GPU_SpGEMMWorkEstimationOrComputeOp : GPU_Op<"spgemm_work_estimation_or_compute", [GPU_AsyncOpInterface]> {
22392239
let summary = "SpGEMM work estimation operation";
22402240
let description = [{
2241-
The `gpu.spgemm_work_estimation_or_compute` is used to call
2241+
The `gpu.spgemm_work_estimation_or_compute` is used to call
22422242
cusparseSpGEMM_workEstimation or cusparseSpGEMM_compute. Both of them are
22432243
for both determining the buffer size and performing the actual computation.
22442244
The operation expects handles returned by previous sparse operations to
2245-
construct an environment and the operands for SpGEMM.
2245+
construct an environment and the operands for SpGEMM.
22462246
The buffer must have been allocated on the device.
22472247

22482248

@@ -2256,8 +2256,8 @@ def GPU_SpGEMMWorkEstimationOrComputeOp : GPU_Op<"spgemm_work_estimation_or_comp
22562256

22572257
```mlir
22582258
%bufferSz, %token = gpu.spgemm_work_estimation_or_compute async [%dep]{COMPUTE}
2259-
%desc, %spmatA{NON_TRANSPOSE}, %spmatB{NON_TRANSPOSE},
2260-
%spmatC, ALG2, %spgemmDesc, %c0, %alloc: f32 into
2259+
%desc, %spmatA{NON_TRANSPOSE}, %spmatB{NON_TRANSPOSE},
2260+
%spmatC, ALG2, %spgemmDesc, %c0, %alloc: f32 into
22612261
memref<0xi8>
22622262
```
22632263

@@ -2358,8 +2358,8 @@ def GPU_SpGEMMEstimateMemoryOp : GPU_Op<"spgemm_estimate_memory", [GPU_AsyncOpIn
23582358
auto modeA = gpu::TransposeMode::NON_TRANSPOSE;
23592359
auto modeB = gpu::TransposeMode::NON_TRANSPOSE;
23602360
auto alg = gpu::SpGEMMAlg::ALG1;
2361-
return build($_builder, $_state, bufferSz3New, bufferSz2New, asyncToken,
2362-
asyncDependencies, desc, modeA, modeB, spmatA, spmatB, spmatC,
2361+
return build($_builder, $_state, bufferSz3New, bufferSz2New, asyncToken,
2362+
asyncDependencies, desc, modeA, modeB, spmatA, spmatB, spmatC,
23632363
computeType, alg, bufferSz3, buffer3, bufferSz2);}]>
23642364
];
23652365

@@ -2375,7 +2375,7 @@ def GPU_SpGEMMCopyOp : GPU_Op<"spgemm_copy", [GPU_AsyncOpInterface]> {
23752375
let description = [{
23762376
The `gpu.spgemm_copy` operation copies a sparse matrix, e.g., the result of
23772377
the SpGEMM computation.
2378-
2378+
23792379
If the `async` keyword is present, the op is executed asynchronously (i.e.
23802380
it does not block until the execution has finished on the device). In
23812381
that case, it returns a `!gpu.async.token` in addition to the environment.
@@ -2430,7 +2430,7 @@ def GPU_SpGEMMGetSizeOp : GPU_Op<"spgemm_get_size", [GPU_AsyncOpInterface]> {
24302430
let description = [{
24312431
The `gpu.spgemm_get_size` operation retrieves the number of rows, number of
24322432
columns, and number of non-zero elements of a sparse matrix.
2433-
2433+
24342434
If the `async` keyword is present, the op is executed asynchronously (i.e.
24352435
it does not block until the execution has finished on the device). In
24362436
that case, it returns a `!gpu.async.token` in addition to the environment.

0 commit comments

Comments
 (0)