Mat2_.is_sparse() INTERNAL ASSERT FAILED at "../aten/src/ATen/native/sparse/cuda/SparseMatMul.cu":786,

Getting this error when trying to run GIN with pytorch lightning