in src/hardware/hardware_detector.py [0:0]
def get_recommended_hardware(self) -> List[str]:
recommended = []
if self.detect_nvidia_gpu():
recommended.append(('cuda', 'NVIDIA GPU detected'))
if self.detect_intel_cpu():
recommended.append(('intel_cpu', 'Intel CPU detected'))
if self.detect_amd_cpu():
recommended.append(('amd_cpu', 'AMD CPU detected'))
if self.detect_intel_gpu():
recommended.append(('intel_gpu', 'Intel GPU detected'))
if self.detect_amd_gpu():
recommended.append(('rocm', 'AMD GPU detected'))
if self.detect_habana():
recommended.append(('habana', 'Habana device detected'))
if self.detect_tpu():
recommended.append(('tpu', 'Google TPU detected'))
if self.detect_inferentia():
recommended.append(('inferentia', 'AWS Inferentia detected'))
if self.detect_apple_silicon():
recommended.append(('apple_silicon', 'Apple Silicon detected'))
return recommended