Home
last modified time | relevance | path

Searched refs:is_gpu_block_index (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Dstmt.h578 bool is_gpu_block_index() const { in is_gpu_block_index() function
587 if (!is_gpu_block_index()) { in gpu_block_index_str()
614 if (is_gpu_block_index() && gpu_block_index() != index) { in set_gpu_block_index()
650 if (is_gpu_block_index()) { in set_gpu_thread_index()
668 if (is_gpu_block_index()) { in ToString()
Dblock_codegen.cpp75 if (loop_options.is_gpu_block_index()) { in visit()
116 if (loop_options.is_gpu_block_index()) { in visit()
Dcuda_codegen.cpp103 if (for_v->loop_options().is_gpu_block_index()) { in visit()
127 if (loop_options.is_gpu_block_index()) { in visit()
672 if (loop_options.is_gpu_block_index()) { in mutate()
712 if (loop_options.is_gpu_block_index()) { in mutate()
Dregisterizer.cpp194 if (v->loop_options().is_gpu_block_index() || in visit()
Dmem_dependency_checker.cpp820 bool parallelized = v->loop_options().is_gpu_block_index() || in visit()
Dloopnest.cpp1485 if (f->loop_options().is_gpu_block_index() || in sliceHead()
1528 if (f->loop_options().is_gpu_block_index() || in sliceTail()
/external/pytorch/test/cpp/tensorexpr/
Dtest_loopnest.cpp178 ASSERT_TRUE(tail->loop_options().is_gpu_block_index()); in TEST()
207 ASSERT_TRUE(tail_head->loop_options().is_gpu_block_index()); in TEST()
866 ASSERT_TRUE(outer->loop_options().is_gpu_block_index()); in TEST()
893 ASSERT_TRUE(outer->loop_options().is_gpu_block_index()); in TEST()