Prune unused defs from pytorch module compilation

microsoft / knossos-ksc

Compiler with automatic differentiation

Other

45 stars 10 forks source link

Closed dcrc2 closed 3 years ago

dcrc2 commented 3 years ago

Makes use of the new ksc option added in #1020.

The generated code for relu3 is reduced from 6965 lines to just 275 (which includes 40 lines of entry points not generated by ksc).

Fixes #1022

awf commented 3 years ago

Yay!! Thanks!

awf commented 3 years ago

Fixes #1022

awf commented 3 years ago

If you get a chance, it would be interesting to know what this does to compile times.