• Nick Stathas's avatar
    Skip unnecessary assertions and enable non-blocking data transfers (#195) · fe8c3ce3
    Nick Stathas authored
    * Uses the `trust_data` invariant to skip blocking assertions, when unnecessary, during construction of `SparseStorage` objects.
    * Refactors the dtype and device transfer APIs to align with `torch.Tensor` while maintaining backward compatibility.
    * No longer constructs dummy tensors when changing dtype or device.
    fe8c3ce3
setup.py 4.4 KB