12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- // Copyright 2018 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
- //go:build (386 || amd64 || amd64p32) && gc
- // +build 386 amd64 amd64p32
- // +build gc
- #include "textflag.h"
- // func cpuid(eaxArg, ecxArg uint32) (eax, ebx, ecx, edx uint32)
- TEXT ·cpuid(SB), NOSPLIT, $0-24
- MOVL eaxArg+0(FP), AX
- MOVL ecxArg+4(FP), CX
- CPUID
- MOVL AX, eax+8(FP)
- MOVL BX, ebx+12(FP)
- MOVL CX, ecx+16(FP)
- MOVL DX, edx+20(FP)
- RET
- // func xgetbv() (eax, edx uint32)
- TEXT ·xgetbv(SB),NOSPLIT,$0-8
- MOVL $0, CX
- XGETBV
- MOVL AX, eax+0(FP)
- MOVL DX, edx+4(FP)
- RET
- // func darwinSupportsAVX512() bool
- TEXT ·darwinSupportsAVX512(SB), NOSPLIT, $0-1
- MOVB $0, ret+0(FP) // default to false
- #ifdef GOOS_darwin // return if not darwin
- #ifdef GOARCH_amd64 // return if not amd64
- // These values from:
- // https://github.com/apple/darwin-xnu/blob/xnu-4570.1.46/osfmk/i386/cpu_capabilities.h
- #define commpage64_base_address 0x00007fffffe00000
- #define commpage64_cpu_capabilities64 (commpage64_base_address+0x010)
- #define commpage64_version (commpage64_base_address+0x01E)
- #define hasAVX512F 0x0000004000000000
- MOVQ $commpage64_version, BX
- CMPW (BX), $13 // cpu_capabilities64 undefined in versions < 13
- JL no_avx512
- MOVQ $commpage64_cpu_capabilities64, BX
- MOVQ $hasAVX512F, CX
- TESTQ (BX), CX
- JZ no_avx512
- MOVB $1, ret+0(FP)
- no_avx512:
- #endif
- #endif
- RET
|