Home
last modified time | relevance | path

Searched defs:device_arch (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBlas.cpp203 std::string device_arch = prop->gcnArchName; in isSupportedHipLtROCmArch() local
865 std::string device_arch = dprops->gcnArchName; in _scaled_mm_allowed_device() local
H A Dint4mm.cu139 std::string device_arch = prop->gcnArchName; in isCDNA2orLater() local
/aosp_15_r20/system/extras/simpleperf/scripts/
H A Dapp_profiler.py65 def __init__(self, ndk_path, device_arch, adb): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp446 std::string device_arch = prop->gcnArchName; in isGPUArch() local