| 62 | |
| 63 | @R.function |
| 64 | def cuda_graph_alloc() -> R.Tuple(R.Object, R.Object): |
| 65 | R.func_attr({"global_symbol": "cuda_graph_alloc"}) |
| 66 | storage: R.Object = R.vm.alloc_storage(R.shape((1024,)), R.prim_value(0), R.dtype("uint8")) |
| 67 | storage1: R.Object = R.vm.alloc_storage(R.shape((1024,)), R.prim_value(0), R.dtype("uint8")) |
| 68 | gv: R.Tuple(R.Object, R.Object) = (storage, storage1) |
| 69 | return gv |
| 70 | |
| 71 | @R.function(pure=False) |
| 72 | def cuda_graph_capture(alloc: R.Tensor((16, 16), dtype="float32"), storage1: R.Object, storage: R.Object) -> R.Tuple(R.Tensor((16, 16), dtype="float32")): |