search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Perf: Handle stride-0 tensors #1060

Open awf opened 2 years ago

awf commented 2 years ago

Today, t.contiguous() copies in the common case that t is the gradient of sum(), i.e. stride-0 tensor of 1s.