Home
last modified time | relevance | path

Searched refs:gpu_thread_index (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcuda_codegen.cpp156 int gpu_thread_index = loop_options.gpu_thread_index(); in visit() local
157 if (gpu_thread_index >= 3) { in visit()
161 if (gpu_thread_extents_.size() <= static_cast<size_t>(gpu_thread_index)) { in visit()
162 gpu_thread_extents_.resize(gpu_thread_index + 1); in visit()
164 prev = gpu_thread_extents_[gpu_thread_index]; in visit()
174 gpu_thread_extents_[gpu_thread_index] = v->stop(); in visit()
178 gpu_thread_extents_[gpu_thread_index] = v->stop(); in visit()
180 gpu_thread_extents_[gpu_thread_index] = in visit()
690 int gpu_thread_index = loop_options.gpu_thread_index(); in mutate() local
691 if (gpu_thread_index >= 3) { in mutate()
[all …]
H A Dstmt.h622 return gpu_thread_index() != IDX_UNSET; in is_gpu_thread_index()
625 int gpu_thread_index() const { in gpu_thread_index() function
653 if (is_gpu_thread_index() && gpu_thread_index() != index) { in set_gpu_thread_index()