microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Use aten constructors in C++ #1017

Open awf opened 2 years ago

awf commented 2 years ago

As per https://pytorch.org/cppdocs/#autograd, the torch::Tensor classes maintain autograd info. This might be necessary for us, or might be wasteful. This issue is to determine if at::empty and related constructors give us our desired behaviour, and if so, to benchmark any change in speed.