search

ROCm / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
219 stars 51 forks source link

Added gcnArchName #1413

Closed BLOrange-AMD closed 4 months ago

BLOrange-AMD commented 4 months ago

This PR is to add gcnArchName in release/2.0 branch.

This is also a follow up PR (https://github.com/ROCm/pytorch/pull/1382) to fix API call and move helper location.

Rebuilt PyTorch and quick tests run fine.