def METHOD_NAME(self):
    def fn(inp, offsets, max_seq_len):
        return torch.ops.test_inductor_ops.jagged_to_padded_dense(
            inp, offsets, max_seq_len, 60.0
        )
    inp = torch.rand((9, 96), device="cuda")
    offsets = torch.tensor([0, 2, 5, 9], dtype=torch.int32, device="cuda")
    max_seq_len = 4
    res = fn(inp, offsets, max_seq_len)
    self.assertEqual(inp[0], res[0][0])
    self.assertEqual(inp[1], res[0][1])
    self.assertEqual(inp[2], res[1][0])
    self.assertEqual(inp[3], res[1][1])
    self.assertEqual(inp[5], res[2][0])
    self.assertEqual(inp[8], res[2][3])
    fn_opt = torch.compile(fn)
    self.assertEqual(
        fn(inp, offsets, max_seq_len), fn_opt(inp, offsets, max_seq_len)
    )