Home
last modified time | relevance | path

Searched refs:gpu_block_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcuda_codegen.cpp128 int gpu_block_index = loop_options.gpu_block_index(); in visit() local
129 if (gpu_block_index >= 3) { in visit()
133 if (gpu_block_extents_.size() <= static_cast<size_t>(gpu_block_index)) { in visit()
134 gpu_block_extents_.resize(gpu_block_index + 1); in visit()
136 prev = gpu_block_extents_[gpu_block_index]; in visit()
146 gpu_block_extents_[gpu_block_index] = v->stop(); in visit()
150 gpu_block_extents_[gpu_block_index] = v->stop(); in visit()
152 gpu_block_extents_[gpu_block_index] = in visit()
673 int gpu_block_index = loop_options.gpu_block_index(); in mutate() local
674 if (gpu_block_index >= 3) { in mutate()
[all …]
H A Dstmt.h582 int gpu_block_index() const { in gpu_block_index() function
614 if (is_gpu_block_index() && gpu_block_index() != index) { in set_gpu_block_index()
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_loopnest.cpp179 ASSERT_EQ(tail->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
208 ASSERT_EQ(tail_head->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
867 ASSERT_EQ(outer->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
894 ASSERT_EQ(outer->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
H A Dtest_simplify.cpp3795 ASSERT_EQ(options.gpu_block_index(), options2.gpu_block_index()); in TEST()