Support `torch.compile` in `segment_matmul` forward - Githubissues

pyg-team / pyg-lib

Low-Level Graph Neural Network Operators for PyG

https://pyg-lib.readthedocs.io

Other

172 stars 42 forks source link

Support `torch.compile` in `segment_matmul` forward #333

Open akihironitta opened 4 months ago

akihironitta commented 4 months ago

Part of https://github.com/pyg-team/pytorch_geometric/issues/8890.

Requires PyTorch 2.4.