e8af2ea63a
* guix/cpu.scm <cpu>: Add vendor field. (current-cpu): Also fill in the 'vendor' field. (cpu->gcc-architecture): Add detection logic for AMD CPUs.
197 lines
8.6 KiB
Scheme
197 lines
8.6 KiB
Scheme
;;; GNU Guix --- Functional package management for GNU
|
|
;;; Copyright © 2021 Ludovic Courtès <ludo@gnu.org>
|
|
;;; Copyright © 2022 Efraim Flashner <efraim@flashner.co.il>
|
|
;;;
|
|
;;; This file is part of GNU Guix.
|
|
;;;
|
|
;;; GNU Guix is free software; you can redistribute it and/or modify it
|
|
;;; under the terms of the GNU General Public License as published by
|
|
;;; the Free Software Foundation; either version 3 of the License, or (at
|
|
;;; your option) any later version.
|
|
;;;
|
|
;;; GNU Guix is distributed in the hope that it will be useful, but
|
|
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
;;; GNU General Public License for more details.
|
|
;;;
|
|
;;; You should have received a copy of the GNU General Public License
|
|
;;; along with GNU Guix. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
(define-module (guix cpu)
|
|
#:use-module (guix sets)
|
|
#:use-module (guix memoization)
|
|
#:use-module (srfi srfi-1)
|
|
#:use-module (srfi srfi-9)
|
|
#:use-module (ice-9 match)
|
|
#:use-module (ice-9 rdelim)
|
|
#:export (current-cpu
|
|
cpu?
|
|
cpu-architecture
|
|
cpu-vendor
|
|
cpu-family
|
|
cpu-model
|
|
cpu-flags
|
|
|
|
cpu->gcc-architecture))
|
|
|
|
;;; Commentary:
|
|
;;;
|
|
;;; This module provides tools to determine the micro-architecture supported
|
|
;;; by the CPU and to map it to a name known to GCC's '-march'.
|
|
;;;
|
|
;;; Code:
|
|
|
|
;; CPU description.
|
|
(define-record-type <cpu>
|
|
(cpu architecture vendor family model flags)
|
|
cpu?
|
|
(architecture cpu-architecture) ;string, from 'uname'
|
|
(vendor cpu-vendor) ;string
|
|
(family cpu-family) ;integer
|
|
(model cpu-model) ;integer
|
|
(flags cpu-flags)) ;set of strings
|
|
|
|
(define current-cpu
|
|
(mlambda ()
|
|
"Return a <cpu> record representing the host CPU."
|
|
(define (prefix? prefix)
|
|
(lambda (str)
|
|
(string-prefix? prefix str)))
|
|
|
|
(call-with-input-file "/proc/cpuinfo"
|
|
(lambda (port)
|
|
(let loop ((vendor #f)
|
|
(family #f)
|
|
(model #f))
|
|
(match (read-line port)
|
|
((? eof-object?)
|
|
#f)
|
|
((? (prefix? "vendor_id") str)
|
|
(match (string-tokenize str)
|
|
(("vendor_id" ":" vendor)
|
|
(loop vendor family model))))
|
|
((? (prefix? "cpu family") str)
|
|
(match (string-tokenize str)
|
|
(("cpu" "family" ":" family)
|
|
(loop vendor (string->number family) model))))
|
|
((? (prefix? "model") str)
|
|
(match (string-tokenize str)
|
|
(("model" ":" model)
|
|
(loop vendor family (string->number model)))
|
|
(_
|
|
(loop vendor family model))))
|
|
((? (prefix? "flags") str)
|
|
(match (string-tokenize str)
|
|
(("flags" ":" flags ...)
|
|
(cpu (utsname:machine (uname))
|
|
vendor family model (list->set flags)))))
|
|
(_
|
|
(loop vendor family model))))))))
|
|
|
|
(define (cpu->gcc-architecture cpu)
|
|
"Return the architecture name, suitable for GCC's '-march' flag, that
|
|
corresponds to CPU, a record as returned by 'current-cpu'."
|
|
(match (cpu-architecture cpu)
|
|
("x86_64"
|
|
;; Transcribed from GCC's 'host_detect_local_cpu' in driver-i386.c.
|
|
(or (and (equal? "GenuineIntel" (cpu-vendor cpu))
|
|
(= 6 (cpu-family cpu)) ;the "Pentium Pro" family
|
|
(letrec-syntax ((if-flags (syntax-rules (=>)
|
|
((_)
|
|
#f)
|
|
((_ (flags ... => name) rest ...)
|
|
(if (every (lambda (flag)
|
|
(set-contains? (cpu-flags cpu)
|
|
flag))
|
|
'(flags ...))
|
|
name
|
|
(if-flags rest ...))))))
|
|
|
|
(if-flags ("avx" "avx512vp2intersect" "tsxldtrk" => "sapphirerapids")
|
|
("avx" "avx512vp2intersect" => "tigerlake")
|
|
("avx" "avx512bf16" => "cooperlake")
|
|
("avx" "wbnoinvd" => "icelake-server")
|
|
("avx" "avx512bitalg" => "icelake-client")
|
|
("avx" "avx512vbmi" => "cannonlake")
|
|
("avx" "avx5124vnniw" => "knm")
|
|
("avx" "avx512er" => "knl")
|
|
("avx" "avx512f" => "skylake-avx512")
|
|
("avx" "serialize" => "alderlake")
|
|
("avx" "clflushopt" => "skylake")
|
|
("avx" "adx" => "broadwell")
|
|
("avx" "avx2" => "haswell")
|
|
("avx" => "sandybridge")
|
|
("sse4_2" "gfni" => "tremont")
|
|
("sse4_2" "sgx" => "goldmont-plus")
|
|
("sse4_2" "xsave" => "goldmont")
|
|
("sse4_2" "movbe" => "silvermont")
|
|
("sse4_2" => "nehalem")
|
|
("ssse3" "movbe" => "bonnell")
|
|
("ssse3" => "core2")
|
|
("longmode" => "x86-64"))))
|
|
|
|
(and (equal? "AuthenticAMD" (cpu-vendor cpu))
|
|
(letrec-syntax ((if-flags (syntax-rules (=>)
|
|
((_)
|
|
#f)
|
|
((_ (flags ... => name) rest ...)
|
|
(if (every (lambda (flag)
|
|
(set-contains? (cpu-flags cpu)
|
|
flag))
|
|
'(flags ...))
|
|
name
|
|
(if-flags rest ...))))))
|
|
|
|
(or (and (= 22 (cpu-family cpu))
|
|
(if-flags ("movbe" => "btver2")))
|
|
(and (= 6 (cpu-family cpu))
|
|
(if-flags ("3dnowp" => "athalon")))
|
|
(if-flags ("vaes" => "znver3")
|
|
("clwb" => "znver2")
|
|
("clzero" => "znver1")
|
|
("avx2" => "bdver4")
|
|
("xsaveopt" => "bdver3")
|
|
("bmi" => "bdver2")
|
|
("xop" => "bdver1")
|
|
("sse4a" "has_ssse3" => "btver1")
|
|
("sse4a" => "amdfam10")
|
|
("sse2" "sse3" => "k8-sse3")
|
|
("longmode" "sse3" => "k8-sse3")
|
|
("sse2" => "k8")
|
|
("longmode" => "k8")
|
|
("mmx" "3dnow" => "k6-3")
|
|
("mmx" => "k6")
|
|
(_ => "pentium")))))
|
|
|
|
;; Fallback case for non-Intel processors or for Intel processors not
|
|
;; recognized above.
|
|
(letrec-syntax ((if-flags (syntax-rules (=>)
|
|
((_)
|
|
#f)
|
|
((_ (flags ... => name) rest ...)
|
|
(if (every (lambda (flag)
|
|
(set-contains? (cpu-flags cpu)
|
|
flag))
|
|
'(flags ...))
|
|
name
|
|
(if-flags rest ...))))))
|
|
(if-flags ("avx512" => "knl")
|
|
("adx" => "broadwell")
|
|
("avx2" => "haswell")
|
|
;; TODO: tigerlake, cooperlake, etc.
|
|
("avx" => "sandybridge")
|
|
("sse4_2" "gfni" => "tremont")
|
|
("sse4_2" "sgx" => "goldmont-plus")
|
|
("sse4_2" "xsave" => "goldmont")
|
|
("sse4_2" "movbe" => "silvermont")
|
|
("sse4_2" => "nehalem")
|
|
("ssse3" "movbe" => "bonnell")
|
|
("ssse3" => "core2")))
|
|
|
|
;; TODO: Recognize CENTAUR/CYRIX/NSC?
|
|
|
|
"x86_64"))
|
|
(architecture
|
|
;; TODO: AArch64.
|
|
architecture)))
|