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"],
|