microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Remove special case for ATen benchmark #1001

Closed dcrc2 closed 2 years ago

dcrc2 commented 2 years ago

Now that we have "embedded" benchmarks which take torch::Tensor arguments, this gives us a convenient way to write benchmarks which use ATen C++ functions. So there is no need for the separate "ATen" benchmark in conftest.py. The new method requires much less boilerplate.