microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Add LM AD transformation for map #748

Open toelli-msft opened 3 years ago

toelli-msft commented 3 years ago

See https://github.com/microsoft/knossos-ksc/pull/727

awf commented 3 years ago

What is required? A linear map lmdiag? We do need this to compare with SUF.

toelli-msft commented 3 years ago

Yes, some sort of linear map primitive to express the derivative of map, as discussed at https://github.com/microsoft/knossos-ksc/issues/714. Then we need to implement it in the backend as well.

This is the same strategy as used for fold.

toelli-msft commented 3 years ago

(My strategy for implementing this will approximately be to copy what I did for fold)