theseus/extlib/mat_mult.cu (9 lines):
	- line 94: TORCH_CHECK(Ms_val.dtype() == torch::kDouble); // TODO: add support for float
	- line 112: // TODO: do experiments on choice of work group size
	- line 123: // TODO: set stream according to torch
	- line 183: TORCH_CHECK(Ms_val.dtype() == torch::kDouble); // TODO: add support for float
	- line 197: // TODO: do experiments on choice of work group size
	- line 258: TORCH_CHECK(Ms_val.dtype() == torch::kDouble); // TODO: add support for float
	- line 272: // TODO: do experiments on choice of work group size
	- line 333: TORCH_CHECK(Ms_val.dtype() == torch::kDouble); // TODO: add support for float
	- line 340: // TODO: do experiments on choice of work group size