Skip to content

Commit 9435723

Browse files
committed
add 'experimental' declaration in dynamo example codes
1 parent bedb7c8 commit 9435723

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

examples/cpu/inference/BERT_torchdynamo_mode_inference_fp32.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
seq_length = 512
1010
data = torch.randint(vocab_size, size=[batch_size, seq_length])
1111

12+
# Experimental Feature
1213
#################### code changes ####################
1314
import intel_extension_for_pytorch as ipex
1415
model = torch.compile(model, backend="ipex")

examples/cpu/inference/resnet50_torchdynamo_mode_inference_fp32.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1-
21
import torch
32
import torchvision.models as models
43

54
model = models.resnet50(weights=models.ResNet50_Weights.DEFAULT)
65
model.eval()
76
data = torch.rand(1, 3, 224, 224)
87

8+
# Experimental Feature
99
#################### code changes ####################
1010
import intel_extension_for_pytorch as ipex
1111
model = torch.compile(model, backend="ipex")

0 commit comments

Comments
 (0)