Comco is an MLIR dialect providing higher level machine learning primitives and
a set of incremental lowerings to MLIR's gpu/nvgpu dialects. Comco also
performs basic kernel fusion on tensor and broadcasting operations
(i.e. GEMM + AllReduce).
-
Notifications
You must be signed in to change notification settings - Fork 1
MLIR dialect for higher level machine learning primitives
License
mattnappo/comco
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
MLIR dialect for higher level machine learning primitives
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published