1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# Owner(s): ["module: inductor"]
#
# This smoketest is referenced in the internal-only minifier runbook
# https://docs.google.com/document/d/18L9e7bZSBpJ7gGbwlUV13LasmjiEX2lree2pl-SdbCU/edit
import os
os.environ["TORCHDYNAMO_REPRO_AFTER"] = "dynamo"
import torch
import torch._dynamo as torchdynamo
import torch._inductor.config
import torch._ops
torch._inductor.config.cpp.inject_relu_bug_TESTING_ONLY = "compile_error"
def func(x):
x = torch.sigmoid(x)
x = torch.mul(x, torch.ones(2))
x = torch.relu(x)
x = torch.add(x, torch.zeros(2))
x = torch.ops.aten.round(x)
return x
def run_internal_minifier():
torchdynamo.config.debug_dir_root = "."
f_opt = torch.compile(func)
f_opt(torch.ones(2))
run_internal_minifier()
|
