(inp)
| 760 | def test_jit_split_simple(self): |
| 761 | @TinyJit |
| 762 | def f(inp): |
| 763 | op0 = self.compute(Device.DEFAULT, inp) |
| 764 | op1 = self.compute(Device.DEFAULT, op0) |
| 765 | op2 = self.compute(Device.DEFAULT, op1) |
| 766 | return op2 |
| 767 | |
| 768 | inp = Tensor.randn(10, 10, device=Device.DEFAULT).realize() |
| 769 | self.expect(f, inp, |