Searched refs:compile_to_sass (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/jit/codegen/fuser/cuda/ |
| D | fused_kernel.cpp | 33 bool& compile_to_sass) { in codegenOutputQuery() argument
36 compile_to_sass = false; in codegenOutputQuery()
74 compile_to_sass = false; in codegenOutputQuery()
78 compile_to_sass = true; in codegenOutputQuery()
115 bool compile_to_sass = false; in FusedKernelCUDA() local
116 codegenOutputQuery(prop_, major, minor, compile_to_sass); in FusedKernelCUDA()
136 (compile_to_sass ? "sm_" : "compute_") + in FusedKernelCUDA()
162 const auto getSize = compile_to_sass in FusedKernelCUDA()
165 const auto getFunc = compile_to_sass in FusedKernelCUDA()
|
| D | fused_kernel.h | 21 bool& compile_to_sass);
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | jit_utils.cpp | 868 bool& compile_to_sass) { in codegenOutputQuery() argument
873 compile_to_sass = false; in codegenOutputQuery()
908 compile_to_sass = false; in codegenOutputQuery()
912 compile_to_sass = true; in codegenOutputQuery()
917 compile_to_sass = false; in codegenOutputQuery()
1477 bool compile_to_sass = false; in jit_pwise_function() local
1479 prop, cuda_major, cuda_minor, nvrtc_major, nvrtc_minor, compile_to_sass); in jit_pwise_function()
1509 ss << (compile_to_sass ? "_sass" : "_ptx"); in jit_pwise_function()
1550 (compile_to_sass ? "sm_" : "compute_") + std::to_string(cuda_major) + in jit_pwise_function()
1582 const auto getSize = compile_to_sass in jit_pwise_function()
[all …]
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | cuda_codegen.cpp | 1276 bool compile_to_sass = false; in CompileToNVRTC() local
1277 fuser::cuda::codegenOutputQuery(prop, major, minor, compile_to_sass); in CompileToNVRTC()
1297 (compile_to_sass ? "sm_" : "compute_") + in CompileToNVRTC()
1327 auto getSize = compile_to_sass in CompileToNVRTC()
1330 auto getFunc = compile_to_sass ? at::globalContext().getNVRTC().nvrtcGetCUBIN in CompileToNVRTC()
|