mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
b9b4f05abf
Summary: Have basic reduction fusion working, and have improved code generator to approach performance of eager mode reductions. Coming soon will be pointwise-reduction fusions in a way that should prevent the possibility of hitting regressions. Also working on performant softmax kernels in the code generator which may be our next fusion target. Pull Request resolved: https://github.com/pytorch/pytorch/pull/40864 Reviewed By: ngimel Differential Revision: D22392877 Pulled By: soumith fbshipit-source-id: 457448a807d628b1035f6d90bc0abe8a87bf8447
7 lines
127 B
Python
7 lines
127 B
Python
import sys
|
|
sys.argv.append("--ge_config=legacy")
|
|
from test_jit_cuda_fuser import *
|
|
|
|
if __name__ == '__main__':
|
|
run_tests()
|