Home
last modified time | relevance | path

Searched defs:has_cuda (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp2169 PyObject* has_cuda = Py_True; in initModule() local
2171 PyObject* has_cuda = Py_False; in initModule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dattention.cpp461 bool has_cuda = query_key_set.has(c10::DispatchKey::CUDA); in _fused_sdp_choice_meta() local
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp2395 bool has_cuda = at::globalContext().hasCUDA(); in TEST() local