Searched defs:xnumel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_heuristics.py | 94 def triton_(in_ptr0, out_ptr0, xnumel, XBLOCK: tl.constexpr): argument
|
| H A D | test_cuda_repro.py | 443 def kernel(in_out_ptr0, in_ptr0, xnumel, XBLOCK: tl.constexpr): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | triton_heuristics.py | 1102 def check_config(cfg, *, xnumel=None, ynumel=None, znumel=None): argument
1728 def split_scan_grid(xnumel, rnumel): argument
|