Searched defs:isGPUArch (Results 1 – 2 of 2) sorted by relevance
190 …virtual bool isGPUArch(DeviceIndex /*device_index*/, const std::vector<std::string>& /*archs*/) co… in isGPUArch() function
444 bool CUDAHooks::isGPUArch(DeviceIndex device_index, const std::vector<std::string>& archs) const { in isGPUArch() function in at::cuda::detail::CUDAHooks