Skip to content

Feat/metax conv softmax#1409

Merged
wooway777 merged 2 commits into
mainfrom
feat/metax-conv-softmax
Jul 23, 2026
Merged

Feat/metax conv softmax#1409
wooway777 merged 2 commits into
mainfrom
feat/metax-conv-softmax

Conversation

@wooway777

Copy link
Copy Markdown
Collaborator

No description provided.

rubik and others added 2 commits July 23, 2026 09:35
Implement convolution and softmax operators for Metax backend.

- Conv: Uses mcDNN library for standard paths plus custom MACA kernels
  for 2D direct conv and 3D patch embed optimizations.
- Softmax: Implements custom kernel using warp/block reduction for
  high throughput.
- Types: Supports FP16, BF16, and FP32 for both operators.
@wooway777
wooway777 requested a review from a team July 23, 2026 11:36
@wooway777
wooway777 merged commit c07b83c into main Jul 23, 2026
8 of 10 checks passed
@wooway777
wooway777 deleted the feat/metax-conv-softmax branch July 23, 2026 11:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant