physicsnemo icon indicating copy to clipboard operation
physicsnemo copied to clipboard

⛰️[EPIC]: FIGConvNet integration for aerodynamic evaluation

Open mnabian opened this issue 1 year ago • 3 comments

### Integration
- [x] Integration of the TPUN
- [x] Enable Hydra config
- [ ] Add docstrings
- [ ] Add license headers
- [ ] Add README
- [ ] Use Modulus Distributed Manager
- [ ] Use Modulus checkpointing
- [ ] Use Modulus logging
### Verification
- [ ] Verify on the DrivAER dataset
- [ ] Verify on the Ahmed body dataset
### Optimization
- [ ] Enable AMP
- [ ] Enable Tensor Core compute
- [ ] Enable CUDA Graphs
- [ ] Enable JIT compilation
- [ ] Profile the code and identify the bottlenecks
- [ ] Other performance optimizations identified in profiles
### Maintenance
- [ ] Develop unit tests
- [ ] Perform QA

mnabian avatar Feb 21 '24 22:02 mnabian

@mnabian What we please add the details to the scope of this epic?

ram-cherukuri avatar Sep 03 '24 19:09 ram-cherukuri

@Alexey-Kamenev do you have another epic for FigConvNet integration? If so, we can close this one as duplicate.

mnabian avatar Sep 12 '24 01:09 mnabian

No, that's the only one. I will rename it to use FIGConvNet instead of TPUN (that's the old name).

Alexey-Kamenev avatar Sep 17 '24 18:09 Alexey-Kamenev