Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.49k stars 480 forks source link

Closed anishfish2 closed 1 month ago

anishfish2 commented 1 month ago

Fixes #8103 and #8104 by adding @op decorators to allow the tests to pass.

Abhishek-More commented 1 month ago

Great work!