Flux support for sparse CuArrays

Hey.

I opened an issue on GitHub in the CuArrays repo but this may be a better avenue. Does Flux support (fully/partially) the sparse arrays implemented in that repo? Are there any hidden errors/issues I should be aware of? Flux docs indicate that it doesn’t mind different array types, but CuSparse seems like a relatively recent addition to the CuArrays repo and might not be fully fleshed out yet.

1 Like