[TorchDISC] To support atten::max_pool2d operator
Open
- 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