Searched refs:tensors_to_clone (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | reinplace.py | 501 tensors_to_clone, argument
510 tensors_to_clone,
522 tensors_to_clone: List[str] = []
573 tensors_to_clone.append(arg)
578 tensors_to_clone,
581 return tensors_to_clone
615 tensors_to_clone, _ = get_mutable_args(_mutable_op)
617 tensors_to_clone = [
618 t for t in tensors_to_clone if node.kwargs[t] is not None
620 tensors_to_clone = reinplace_and_refine_tensors_to_clone(
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | triton_kernel_wrap.py | 545 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone): argument
551 tensors_to_clone=tensors_to_clone,
638 tensors_to_clone = identify_mutated_tensors(
647 tensors_to_clone=tensors_to_clone,
667 *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
674 key: (clone_preserve_strides(val) if key in tensors_to_clone else val)
683 return {key: val for key, val in kwargs.items() if key in tensors_to_clone}
688 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
698 if key in tensors_to_clone
704 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_kernels.py | 110 tensors_to_clone=["in_ptr0", "in_ptr1", "out_ptr"],
139 tensors_to_clone=["in_ptr0", "out_ptr"],
|