def METHOD_NAME(self, device, dtypes, shape_strides):
    a_buffer = torch.rand(9, device=device).mul(10).type(dtypes[0])
    b_buffer = torch.rand(9, device=device).mul(10).type(dtypes[1])
    a = a_buffer.as_strided(*shape_strides[0])
    b = b_buffer.as_strided(*shape_strides[1])
    expected = ref_fn(a, b)
    result = jitted_fn(a, b)
    self.assertEqual(expected, result)