code: def test_ckpt_inputs2_outputs1(self, mask):
      module = MaskedLinear(HIDDEN_DIM, HIDDEN_DIM)
      inputs = torch.rand(HIDDEN_DIM)
      inputs.requires_grad = True
      _test_activation_checkpoint(module, inputs, mask)
  
imports: import pytest
  import torch
  import deepspeed
  from copy import deepcopy
  from unit.common import DistributedTest
label_desc: Use image node_id 2 for calling the TestActivationCheckpoint obj's underlying member method code with example usage: obj.test_ckpt_inputs2_outputs1(mask) without return types