Home
last modified time | relevance | path

Searched refs:set_gpu_block_index (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_cuda.cpp49 loops[1]->set_gpu_block_index(0); in testCudaTestVectorAdd01_impl()
117 loops[1]->set_gpu_block_index(0); in TEST()
180 loops[0]->set_gpu_block_index(0); in testCudaTestVectorAdd02_impl()
346 loops[1]->set_gpu_block_index(0); in TEST()
399 loops[0]->set_gpu_block_index(0); in TEST()
468 block_idx_options.set_gpu_block_index(0); in TEST()
547 block_idx_options.set_gpu_block_index(0); in TEST()
638 block_idx_options.set_gpu_block_index(0); in TEST()
701 block_idx_opt.set_gpu_block_index(0); in TEST()
836 block_idx_opt.set_gpu_block_index(0); in TEST()
[all …]
H A Dtest_loopnest.cpp172 loops[0]->set_gpu_block_index(LoopOptions::IDX_Y); in TEST()
201 tail->set_gpu_block_index(LoopOptions::IDX_Y); in TEST()
859 loops[0]->set_gpu_block_index(LoopOptions::IDX_Y); in TEST()
888 loops[0]->set_gpu_block_index(LoopOptions::IDX_Y); in TEST()
H A Dtest_simplify.cpp495 for_stmt->set_gpu_block_index(LoopOptions::IDX_X); in TEST()
501 for_stmt->set_gpu_block_index(LoopOptions::IDX_UNSET); in TEST()
3789 options.set_gpu_block_index(LoopOptions::IDX_W); in TEST()
H A Dtest_reductions.cpp662 loops[0]->set_gpu_block_index(0); in TEST()
H A Dtest_registerizer.cpp965 loopOpts.set_gpu_block_index(0); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dstmt.h606 void set_gpu_block_index(int index) { in set_gpu_block_index() function
775 void set_gpu_block_index(int block_index) { in set_gpu_block_index() function
776 loop_options_.set_gpu_block_index(block_index); in set_gpu_block_index()
H A Dkernel.cpp869 flattened->set_gpu_block_index(0); in transformLoops()
881 inner->set_gpu_block_index(0); in transformLoops()
911 flattened->set_gpu_block_index(0); in transformLoops()
H A Dtensorexpr_init.cpp417 self.set_gpu_block_index(block_index); in initTensorExprBindings()