alibaba / alibaba/BladeDISC

[TorchDISC] To support atten::max_pool2d operator

Open
#266 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
C++
Stars
933
Forks
169
PR merge metrics
No merged PRs in 30d

Description

The atten::max_pool2d operator is not supported by disc. This leads to fragmented disc clusters on some CNN models, e.g. VGG, CRNN.
If atten::max_pool2d operator gets supported, the performance of disc on some CNN models (especially with fp16 AMP) will get improved significantly.

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue identifies the atten::max_pool2d operator as unsupported in DISC and cites VGG and CRNN as affected models. Investigate the TorchDISC operator-support path and verify that these models no longer produce fragmented DISC clusters, using improved performance as the completion signal.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, pytorch
Domain
compilers, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.