ORNL / HydraGNN

Distributed PyTorch implementation of multi-headed graph convolutional neural networks
BSD 3-Clause "New" or "Revised" License
54 stars 24 forks source link

Add cudasync #247

Closed jychoi-hpc closed 1 month ago

jychoi-hpc commented 1 month ago

This update includes:

Ref: https://www.speechmatics.com/company/articles-and-news/timing-operations-in-pytorch

allaffa commented 1 month ago

@jychoi-hpc Black formatting needs to be fixed.