Skip to content

GitLab

"references/vscode:/vscode.git/clone" did not exist on "9d7177fe707b996a90c1071f2e9bc1e7e5079ce3"
Unverified Commit 12f83480 authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

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

parent c5e683c4
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
examples/sampling/graphbolt/pyg/node_classification_advanced.py
View file @ 12f83480
......@@ -54,6 +54,10 @@ import time
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment