Skip to content

GitLab

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
Hide 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 ...@@ -54,6 +54,10 @@ import time
import dgl.graphbolt as gb import dgl.graphbolt as gb
import torch 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 torch.nn.functional as F
import torchmetrics.functional as MF import torchmetrics.functional as MF
from torch_geometric.nn import SAGEConv 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