Skip to content

[Feature]: Implement All Gather + GEMM Fused Kernels #74

@octatrifan-amd

Description

@octatrifan-amd

Suggestion Description

Implement one kernel that fuses the all gather and the GEMM computation or fused MLP, collective use-case.

Operating System

No response

GPU

No response

ROCm Component

No response

Metadata

Metadata

Labels

enhancementNew feature or requestexamplesExamples showcasing Iris APIs and usage

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions