[GraphBolt] PyG advanced example torch.compile() bug workaround.

dmlc · Apr 3, 2024 · ac097d1 · ac097d1
1 parent b743cde
commit ac097d1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/sampling/graphbolt/pyg/node_classification_advanced.py b/examples/sampling/graphbolt/pyg/node_classification_advanced.py
@@ -54,6 +54,9 @@
 
 import dgl.graphbolt as gb
 import torch
+# Needed until https://github.com/pytorch/pytorch/issues/121197 is resolved to
+# use the `--torch-compile` cmdline option reliably.
+import torch._inductor.codecache
 import torch.nn.functional as F
 import torchmetrics.functional as MF
 from torch_geometric.nn import SAGEConv