53 lines
1.5 KiB
ArmAsm
53 lines
1.5 KiB
ArmAsm
|
// 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
|