Closed Roger-luo closed 4 years ago
struct Foo a::Int b::Int end @device function (m::Foo)(theta) @ctrl m.a m.b=>shift(theta) end