Home
last modified time | relevance | path

Searched refs:set_gpu_thread_index (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_cuda.cpp50 loops[2]->set_gpu_thread_index(0); in testCudaTestVectorAdd01_impl()
118 loops[2]->set_gpu_thread_index(0); in TEST()
181 n_inner->set_gpu_thread_index(0); in testCudaTestVectorAdd02_impl()
347 loops[2]->set_gpu_thread_index(0); in TEST()
400 inner->set_gpu_thread_index(0); in TEST()
463 thread_idx_options.set_gpu_thread_index(0); in TEST()
532 thread_idx_options.set_gpu_thread_index(0); in TEST()
621 thread_idx_options.set_gpu_thread_index(0); in TEST()
699 thread_idx_opt.set_gpu_thread_index(0); in TEST()
834 thread_idx_opt.set_gpu_thread_index(0); in TEST()
[all …]
H A Dtest_reductions.cpp663 loops[1]->set_gpu_thread_index(0); in TEST()
H A Dtest_simplify.cpp506 for_stmt->set_gpu_thread_index(LoopOptions::IDX_X); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dstmt.h645 void set_gpu_thread_index(int index) { in set_gpu_thread_index() function
779 void set_gpu_thread_index(int thread_index) { in set_gpu_thread_index() function
780 loop_options_.set_gpu_thread_index(thread_index); in set_gpu_thread_index()
H A Dkernel.cpp870 inner->set_gpu_thread_index(0); in transformLoops()
882 inner1->set_gpu_thread_index(0); in transformLoops()
912 inner->set_gpu_thread_index(0); in transformLoops()
H A Dtensorexpr_init.cpp422 self.set_gpu_thread_index(thread_index); in initTensorExprBindings()