From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: by sourceware.org (Postfix, from userid 7888) id E5E443858028; Thu, 22 Sep 2022 06:50:22 +0000 (GMT) DKIM-Filter: OpenDKIM Filter v2.11.0 sourceware.org E5E443858028 DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gcc.gnu.org; s=default; t=1663829422; bh=limPVo5yB8M/LE3Qbwl6Xv9Hqmev7dN2SHO97doabQw=; h=From:To:Subject:Date:From; b=CmAoUUkxXS82/MlAwZ1RW0myiLjUxQCc3fjho8aK7Ksd9hF0kR2e4eGHhmihOPTzW f6mMO2skBtym/K/1ZeoGIojzl0hihPh/yG393BIV1vt+5gk1j0vY7fiEZrWkGwlItc 8cd75HVWMSRQNdYEFnU6eYfkuktv1RPEoLr/o6EE= MIME-Version: 1.0 Content-Transfer-Encoding: 7bit Content-Type: text/plain; charset="utf-8" From: Haochen Jiang To: gcc-cvs@gcc.gnu.org Subject: [gcc r12-8781] i386: Add syscall to enable AMX for latest kernels X-Act-Checkin: gcc X-Git-Author: Haochen Jiang X-Git-Refname: refs/heads/releases/gcc-12 X-Git-Oldrev: 2b17f9c0e3a1bc36a4f249da5b8b86be20361a3f X-Git-Newrev: dc4cf9fe202b9813cf2a68aeff7cc00cf882aa7d Message-Id: <20220922065022.E5E443858028@sourceware.org> Date: Thu, 22 Sep 2022 06:50:22 +0000 (GMT) List-Id: https://gcc.gnu.org/g:dc4cf9fe202b9813cf2a68aeff7cc00cf882aa7d commit r12-8781-gdc4cf9fe202b9813cf2a68aeff7cc00cf882aa7d Author: Haochen Jiang Date: Thu Jun 16 00:15:53 2022 -0700 i386: Add syscall to enable AMX for latest kernels gcc/testsuite/ChangeLog: * gcc.target/i386/amx-check.h (request_perm_xtile_data): New function to check if AMX is usable and enable AMX. (main): Run test if AMX is usable. Diff: --- gcc/testsuite/gcc.target/i386/amx-check.h | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/gcc/testsuite/gcc.target/i386/amx-check.h b/gcc/testsuite/gcc.target/i386/amx-check.h index 434b0e59703..6fff5ff4631 100644 --- a/gcc/testsuite/gcc.target/i386/amx-check.h +++ b/gcc/testsuite/gcc.target/i386/amx-check.h @@ -4,11 +4,24 @@ #include #include #include +#include +#ifdef __linux__ +#include +#endif #ifdef DEBUG #include #endif #include "cpuid.h" +#define XFEATURE_XTILECFG 17 +#define XFEATURE_XTILEDATA 18 +#define XFEATURE_MASK_XTILECFG (1 << XFEATURE_XTILECFG) +#define XFEATURE_MASK_XTILEDATA (1 << XFEATURE_XTILEDATA) +#define XFEATURE_MASK_XTILE (XFEATURE_MASK_XTILECFG | XFEATURE_MASK_XTILEDATA) + +#define ARCH_GET_XCOMP_PERM 0x1022 +#define ARCH_REQ_XCOMP_PERM 0x1023 + /* TODO: The tmm emulation is temporary for current AMX implementation with no tmm regclass, should be changed in the future. */ @@ -44,6 +57,20 @@ typedef struct __tile /* Stride (colum width in byte) used for tileload/store */ #define _STRIDE 64 +#ifdef __linux__ +/* We need syscall to use amx functions */ +int request_perm_xtile_data() +{ + unsigned long bitmask; + + if (syscall (SYS_arch_prctl, ARCH_REQ_XCOMP_PERM, XFEATURE_XTILEDATA) || + syscall (SYS_arch_prctl, ARCH_GET_XCOMP_PERM, &bitmask)) + return 0; + + return (bitmask & XFEATURE_MASK_XTILE) != 0; +} +#endif + /* Initialize tile config by setting all tmm size to 16x64 */ void init_tile_config (__tilecfg_u *dst) { @@ -185,6 +212,9 @@ main () #endif #ifdef AMX_BF16 && __builtin_cpu_supports ("amx-bf16") +#endif +#ifdef __linux__ + && request_perm_xtile_data () #endif ) {