Searched defs:cuda_cg (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_cuda.cpp | 53 CudaCodeGen cuda_cg(stmt, c, a_buf, b_buf); in testCudaTestVectorAdd01_impl() local
121 CudaCodeGen cuda_cg(stmt, c, a_buf); in TEST() local
184 CudaCodeGen cuda_cg(stmt, c, a_buf, b_buf); in testCudaTestVectorAdd02_impl() local
350 CudaCodeGen cuda_cg(stmt, c); in TEST() local
472 CudaCodeGen cuda_cg(block_idx_loop, data_buf, output_buf); in TEST() local
551 CudaCodeGen cuda_cg(block_idx_loop, a_buf, b_buf); in TEST() local
642 CudaCodeGen cuda_cg(block_idx_loop, a_buf, b_buf); in TEST() local
764 CudaCodeGen cuda_cg(loop_k1, a, b); in TEST() local
894 CudaCodeGen cuda_cg(loop_k, a, b); in TEST() local
1118 CudaCodeGen cuda_cg(loop, a, b, c); in TEST() local
[all …]
|