Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/cuda/
H A Dfused_kernel.cpp29 void codegenOutputQuery( in codegenOutputQuery() function
116 codegenOutputQuery(prop_, major, minor, compile_to_sass); in FusedKernelCUDA()
H A Dfused_kernel.h17 TORCH_CUDA_CU_API void codegenOutputQuery(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.cpp862 void codegenOutputQuery( in codegenOutputQuery() function
1478 at::cuda::jit::codegenOutputQuery( in jit_pwise_function()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcuda_codegen.cpp1277 fuser::cuda::codegenOutputQuery(prop, major, minor, compile_to_sass); in CompileToNVRTC()