Added AMD GPU Support to Zeus

ml-energy / zeus

Deep Learning Energy Measurement and Optimization

https://ml.energy/zeus

Apache License 2.0

179 stars 24 forks source link

Closed parthraut closed 2 months ago

parthraut commented 2 months ago

Added AMD GPU support to Zeus. Involved adding method implementations to Zeus.device.gpu for the AMDGPU class.

jaywonchung commented 2 months ago

Thanks for your work. Please resolve merge conflicts and make CI pass, and then request review.

parthraut commented 2 months ago

@jaywonchung fixed all merge conflicts and passing all tests, ready for review.

parthraut commented 2 months ago

@jaywonchung I went through and retested each method, and fixed any issues. It should be all correct now.