Searched refs:loop_options (Results 1 – 13 of 13) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | block_codegen.cpp | 74 const LoopOptions& loop_options = v->loop_options(); in visit() local
75 if (loop_options.is_gpu_block_index()) { in visit()
76 map_input_to_tensor_bufs_ = loop_options.get_buffer_mapping(); in visit()
78 } else if (loop_options.is_gpu_thread_index()) { in visit()
109 const LoopOptions& loop_options = v->loop_options(); in visit() local
116 if (loop_options.is_gpu_block_index()) { in visit()
137 } else if (loop_options.is_gpu_thread_index()) { in visit()
|
| D | cuda_codegen.cpp | 103 if (for_v->loop_options().is_gpu_block_index()) { in visit()
108 } else if (for_v->loop_options().is_gpu_thread_index()) { in visit()
126 const LoopOptions& loop_options = v->loop_options(); in visit() local
127 if (loop_options.is_gpu_block_index()) { in visit()
128 int gpu_block_index = loop_options.gpu_block_index(); in visit()
155 } else if (loop_options.is_gpu_thread_index()) { in visit()
156 int gpu_thread_index = loop_options.gpu_thread_index(); in visit()
671 const LoopOptions& loop_options = v->loop_options(); in mutate() local
672 if (loop_options.is_gpu_block_index()) { in mutate()
673 int gpu_block_index = loop_options.gpu_block_index(); in mutate()
[all …]
|
| D | stmt.h | 727 const LoopOptions& loop_options) { in make() argument
732 var.node(), start.node(), stop.node(), body, loop_options); in make()
734 const LoopOptions loop_options() const { in loop_options() function
752 LoopOptions loop_options) in For() argument
756 loop_options_(std::move(loop_options)) { in For()
|
| D | ir_mutator.cpp | 333 LoopOptions loop_options = v->loop_options(); in mutate() local
|
| D | registerizer.cpp | 194 if (v->loop_options().is_gpu_block_index() || in visit()
195 v->loop_options().is_gpu_thread_index()) { in visit()
|
| D | loopnest.cpp | 483 LoopOptions loop_options = v->loop_options(); in mutate() local
501 return alloc<For>(var, new_start, new_stop, new_body, loop_options); in mutate()
1485 if (f->loop_options().is_gpu_block_index() || in sliceHead()
1486 f->loop_options().is_gpu_thread_index()) { in sliceHead()
1528 if (f->loop_options().is_gpu_block_index() || in sliceTail()
1529 f->loop_options().is_gpu_thread_index()) { in sliceTail()
|
| D | ir_simplifier.cpp | 1653 LoopOptions loop_options = v->loop_options(); in mutate() local
1662 if (loop_options.isDefault() && loops->isConstant()) { in mutate()
2542 LoopOptions loop_options = v->loop_options(); in mutate() local
2578 if (loop_options.isDefault() && iters->isConstant()) { in mutate()
|
| D | hash_provider.cpp | 201 hash = hash_combine(hash, v->loop_options().ToString()); in visit()
|
| D | ir_cloner.cpp | 256 return alloc<For>(v->var(), start_new, stop_new, body_new, v->loop_options()); in mutate()
|
| D | mem_dependency_checker.cpp | 820 bool parallelized = v->loop_options().is_gpu_block_index() || in visit()
821 v->loop_options().is_gpu_thread_index(); in visit()
|
| D | ir_printer.cpp | 440 std::string loop_options_str = v->loop_options().ToString(); in visit()
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_loopnest.cpp | 178 ASSERT_TRUE(tail->loop_options().is_gpu_block_index()); in TEST()
179 ASSERT_EQ(tail->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
181 ASSERT_TRUE(head->loop_options().isDefault()); in TEST()
207 ASSERT_TRUE(tail_head->loop_options().is_gpu_block_index()); in TEST()
208 ASSERT_EQ(tail_head->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
210 ASSERT_TRUE(head->loop_options().isDefault()); in TEST()
211 ASSERT_TRUE(tail_tail->loop_options().isDefault()); in TEST()
866 ASSERT_TRUE(outer->loop_options().is_gpu_block_index()); in TEST()
867 ASSERT_EQ(outer->loop_options().gpu_block_index(), LoopOptions::IDX_Y); in TEST()
870 ASSERT_TRUE(inner->loop_options().isDefault()); in TEST()
[all …]
|
| D | test_simplify.cpp | 3794 LoopOptions options2 = for_->loop_options(); in TEST()
|