Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.cpp446 std::string device_arch = prop->gcnArchName; in isGPUArch() local
/external/pytorch/aten/src/ATen/native/cuda/
DBlas.cpp203 std::string device_arch = prop->gcnArchName; in isSupportedHipLtROCmArch() local
865 std::string device_arch = dprops->gcnArchName; in _scaled_mm_allowed_device() local
Dint4mm.cu139 std::string device_arch = prop->gcnArchName; in isCDNA2orLater() local