Skip to content

Feat/metax conv#1408

Closed
wooway777 wants to merge 2 commits into
mainfrom
feat/metax-conv
Closed

Feat/metax conv#1408
wooway777 wants to merge 2 commits into
mainfrom
feat/metax-conv

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:32
@wooway777 wooway777 closed this Jul 23, 2026
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