pytorch
fa393b56 - [static runtime] use NNC to generate logit, relu and tanh (#52322)

Commit
3 years ago
[static runtime] use NNC to generate logit, relu and tanh (#52322) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52322 diff BS=1 ``` C2 run finished. Milliseconds per iter: 0.0564008. Iters per second: 17730.3 PyTorch run finished. Milliseconds per iter: 0.0677778. Iters per second: 14754.1 ``` diff BS=20 ``` C2 run finished. Milliseconds per iter: 0.51086. Iters per second: 1957.48 PyTorch run finished. Milliseconds per iter: 0.510077. Iters per second: 1960.49 ``` master BS=1 ``` C2 run finished. Milliseconds per iter: 0.0567362. Iters per second: 17625.4 PyTorch run finished. Milliseconds per iter: 0.0706478. Iters per second: 14154.7 ``` master BS=20 ``` C2 run finished. Milliseconds per iter: 0.510943. Iters per second: 1957.17 PyTorch run finished. Milliseconds per iter: 0.516338. Iters per second: 1936.72 ``` Reviewed By: bertmaher Differential Revision: D25407106 fbshipit-source-id: 08595ba5e4be59e2ef95fb9b24da7e7671692395
Author
Parents
Loading