Aart Bik ee42e23614 [mlir][sparse][gpu] first implementation of the GPU libgen approach ...

The sparse compiler now has two prototype strategies for GPU acceleration:

* CUDA codegen: this converts sparsified code to CUDA threads
* CUDA libgen: this converts pre-sparsified code to cuSPARSE library calls

This revision introduces the first steps required for the second approach.

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D150170

2023-05-15 08:49:38 -07:00

..

BufferizableOpInterfaceImpl.cpp

[mlir][sparse] extend unpack operation to support unpacking a batched COO type

2023-05-01 18:17:29 +00:00

CMakeLists.txt

[mlir][sparse][gpu] a first prototype sparse GPU code generator

2023-04-05 11:32:06 -07:00

CodegenEnv.cpp

[mlir][sparse] Preliminary code changes for ExprId, LatPointId, LatSetId newtypes

2023-03-29 18:01:56 -07:00

CodegenEnv.h

[mlir][sparse] fix build error.

2023-05-04 17:30:13 +00:00

CodegenUtils.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

CodegenUtils.h

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

LoopEmitter.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

LoopEmitter.h

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseBufferRewriting.cpp

[mlir][sparse] Modify the pivot selection method for quick sort.

2023-03-15 13:53:00 -07:00

SparseGPUCodegen.cpp

[mlir][sparse][gpu] first implementation of the GPU libgen approach

2023-05-15 08:49:38 -07:00

SparseStorageSpecifierToLLVM.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseTensorCodegen.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseTensorConversion.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseTensorPasses.cpp

[mlir][sparse][gpu] first implementation of the GPU libgen approach

2023-05-15 08:49:38 -07:00

SparseTensorRewriting.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseTensorStorageLayout.cpp

[mlir][sparse] support iteration over compressed-hi dimension level in loop emitter

2023-04-20 00:57:08 +00:00

SparseTensorStorageLayout.h

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparseVectorization.cpp

[mlir] Update method cast calls to function calls

2023-05-12 11:21:30 +02:00

Sparsification.cpp

[mlir] Move casting calls from methods to function calls

2023-05-12 11:21:25 +02:00

SparsificationAndBufferizationPass.cpp

[mlir][sparse][gpu] first implementation of the GPU libgen approach

2023-05-15 08:49:38 -07:00