Skip to content

feat: add AdaptiveAvgPool3d module (#4832)#5115

Open
jaweed3 wants to merge 2 commits into
tracel-ai:mainfrom
jaweed3:feat/adaptive-avg-pool3d-4832
Open

feat: add AdaptiveAvgPool3d module (#4832)#5115
jaweed3 wants to merge 2 commits into
tracel-ai:mainfrom
jaweed3:feat/adaptive-avg-pool3d-4832

Conversation

@jaweed3

@jaweed3 jaweed3 commented Jun 28, 2026

Copy link
Copy Markdown
Contributor

Implements the 3D adaptive average pooling operation, closing #4832.

Added

  • AdaptiveAvgPool3d module in burn-nn
  • tensor::module::adaptive_avg_pool3d public API
  • ModuleOps::adaptive_avg_pool3d + adaptive_avg_pool3d_backward trait methods
  • IR structs, enum variants, builder macros, and visitor impls
  • Backend implementations: ndarray (native Array5), cubecl (CPU fallback), tch (native), candle (panic), flex (all 4 dtypes), autodiff (backward with checkpointing), fusion, router, dispatch
  • Forward tests (3) and autodiff backward tests (4)

@jaweed3
jaweed3 force-pushed the feat/adaptive-avg-pool3d-4832 branch 5 times, most recently from 12611bd to fd7d7d0 Compare July 7, 2026 12:30
@codecov

codecov Bot commented Jul 7, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 60.00000% with 272 lines in your changes missing coverage. Please review.
✅ Project coverage is 65.91%. Comparing base (724bbb1) to head (718a208).
⚠️ Report is 11 commits behind head on main.

Files with missing lines Patch % Lines
crates/burn-cubecl/src/kernel/pool/base.rs 0.00% 110 Missing ⚠️
crates/burn-fusion/src/ops/module.rs 0.00% 40 Missing ⚠️
crates/burn-router/src/ops/module.rs 0.00% 25 Missing ⚠️
crates/burn-ir/src/operation.rs 0.00% 21 Missing ⚠️
crates/burn-router/src/interpreter.rs 0.00% 13 Missing ⚠️
crates/burn-fusion/src/stream/context.rs 0.00% 12 Missing ⚠️
crates/burn-cubecl/src/ops/module.rs 0.00% 9 Missing ⚠️
crates/burn-tch/src/ops/module.rs 0.00% 8 Missing ⚠️
crates/burn-autodiff/src/ops/module.rs 76.66% 7 Missing ⚠️
crates/burn-candle/src/ops/module.rs 0.00% 7 Missing ⚠️
... and 3 more

❌ Your patch check has failed because the patch coverage (60.00%) is below the target coverage (80.00%). You can increase the patch coverage or adjust the target coverage.
❌ Your project check has failed because the head coverage (65.91%) is below the target coverage (80.00%). You can increase the head coverage or adjust the target coverage.

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #5115      +/-   ##
==========================================
+ Coverage   65.74%   65.91%   +0.16%     
==========================================
  Files        1182     1197      +15     
  Lines      179889   182832    +2943     
==========================================
+ Hits       118275   120520    +2245     
- Misses      61614    62312     +698     

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@jaweed3
jaweed3 force-pushed the feat/adaptive-avg-pool3d-4832 branch 6 times, most recently from e322642 to 412eebd Compare July 11, 2026 16:37
@jaweed3
jaweed3 force-pushed the feat/adaptive-avg-pool3d-4832 branch from 412eebd to 718a208 Compare July 11, 2026 16:40
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