CheckSparseTensorInvariants Class — pytorch Architecture

Source Code

aten/src/ATen/SparseCsrTensorUtils.h lines 150–168

class CheckSparseTensorInvariants {
  bool old_state;

 public:
  CheckSparseTensorInvariants(bool state)
      : old_state(at::globalContext().checkSparseTensorInvariants()) {
    at::globalContext().setCheckSparseTensorInvariants(state);
  }
  CheckSparseTensorInvariants(CheckSparseTensorInvariants&& other) = delete;
  CheckSparseTensorInvariants(const CheckSparseTensorInvariants&) = delete;
  CheckSparseTensorInvariants& operator=(const CheckSparseTensorInvariants&) =
      delete;
  CheckSparseTensorInvariants& operator=(CheckSparseTensorInvariants&&) =
      delete;

  ~CheckSparseTensorInvariants() {
    at::globalContext().setCheckSparseTensorInvariants(old_state);
  }
};