mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
517aee5369
Test Plan:
test script:
```
def test_zhxchen17(self):
from libfb.py.pyinit import initFacebook
initFacebook()
class M(torch.nn.Module):
def forward(self, x):
return torch.add(x, x)
def tmptmp(x, y):
return torch.mul(x, y)
m = M()
n = torch.jit.script(m)
print(n(torch.tensor(1)))
print(torch.jit.script(tmptmp)(torch.tensor(1), torch.tensor(2)))
```
```
I0802 12:01:23.932929 4079081 init.cc:407] Logging to scuba: run __torch__.caffe2.test.export.test_export.M.forward sample rate: 1000000
```
Differential Revision: D60920867
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133484
Approved by: https://github.com/davidberard98