================ @@ -0,0 +1,1017 @@ +//===--- BuiltinsAMDGPU.td - AMDGPU Builtin function defs -------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This file defines the AMDGPU-specific builtin function database. +// +//===----------------------------------------------------------------------===// + +include "clang/Basic/BuiltinsBase.td" + +//===----------------------------------------------------------------------===// +// AMDGPU builtin base classes +//===----------------------------------------------------------------------===// + +class AMDGPUBuiltin<string prototype, list<Attribute> Attr = [], string Feat = ""> : TargetBuiltin { + let Spellings = [NAME]; + let Prototype = prototype; + let Attributes = !listconcat([NoThrow], Attr); + let Features = Feat; +} + +//===----------------------------------------------------------------------===// +// SI+ only builtins. +//===----------------------------------------------------------------------===// + +def __builtin_amdgcn_dispatch_ptr : AMDGPUBuiltin<"void address_space<4> *()", [Const]>; +def __builtin_amdgcn_kernarg_segment_ptr : AMDGPUBuiltin<"void address_space<4> *()", [Const]>; +def __builtin_amdgcn_implicitarg_ptr : AMDGPUBuiltin<"void address_space<4> *()", [Const]>; +def __builtin_amdgcn_queue_ptr : AMDGPUBuiltin<"void address_space<4> *()", [Const]>; + +def __builtin_amdgcn_workgroup_id_x : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_workgroup_id_y : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_workgroup_id_z : AMDGPUBuiltin<"unsigned int()", [Const]>; + +def __builtin_amdgcn_cluster_id_x : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_id_y : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_id_z : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; + +def __builtin_amdgcn_cluster_workgroup_id_x : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_id_y : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_id_z : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_flat_id : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; + +def __builtin_amdgcn_cluster_workgroup_max_id_x : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_max_id_y : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_max_id_z : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; +def __builtin_amdgcn_cluster_workgroup_max_flat_id : AMDGPUBuiltin<"unsigned int()", [Const], "gfx1250-insts">; + +def __builtin_amdgcn_workitem_id_x : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_workitem_id_y : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_workitem_id_z : AMDGPUBuiltin<"unsigned int()", [Const]>; + +def __builtin_amdgcn_workgroup_size_x : AMDGPUBuiltin<"unsigned short()", [Const]>; +def __builtin_amdgcn_workgroup_size_y : AMDGPUBuiltin<"unsigned short()", [Const]>; +def __builtin_amdgcn_workgroup_size_z : AMDGPUBuiltin<"unsigned short()", [Const]>; + +def __builtin_amdgcn_grid_size_x : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_grid_size_y : AMDGPUBuiltin<"unsigned int()", [Const]>; +def __builtin_amdgcn_grid_size_z : AMDGPUBuiltin<"unsigned int()", [Const]>; + +def __builtin_amdgcn_mbcnt_hi : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_mbcnt_lo : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int)", [Const]>; + +def __builtin_amdgcn_s_memtime : AMDGPUBuiltin<"uint64_t()", [], "s-memtime-inst">; + +//===----------------------------------------------------------------------===// +// Instruction builtins. +//===----------------------------------------------------------------------===// +def __builtin_amdgcn_s_getreg : AMDGPUBuiltin<"unsigned int(_Constant int)">; +def __builtin_amdgcn_s_setreg : AMDGPUBuiltin<"void(_Constant int, unsigned int)">; +def __builtin_amdgcn_s_getpc : AMDGPUBuiltin<"uint64_t()">; +def __builtin_amdgcn_s_waitcnt : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_s_sendmsg : AMDGPUBuiltin<"void(_Constant int, unsigned int)">; +def __builtin_amdgcn_s_sendmsghalt : AMDGPUBuiltin<"void(_Constant int, unsigned int)">; +def __builtin_amdgcn_s_barrier : AMDGPUBuiltin<"void()">; +def __builtin_amdgcn_s_ttracedata : AMDGPUBuiltin<"void(int)">; +def __builtin_amdgcn_wave_barrier : AMDGPUBuiltin<"void()">; +def __builtin_amdgcn_sched_barrier : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_sched_group_barrier : AMDGPUBuiltin<"void(_Constant int, _Constant int, _Constant int)">; +def __builtin_amdgcn_iglp_opt : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_s_dcache_inv : AMDGPUBuiltin<"void()">; +def __builtin_amdgcn_buffer_wbinvl1 : AMDGPUBuiltin<"void()">; +def __builtin_amdgcn_fence : AMDGPUBuiltin<"void(unsigned int, char const *, ...)">; +def __builtin_amdgcn_groupstaticsize : AMDGPUBuiltin<"unsigned int()">; +def __builtin_amdgcn_wavefrontsize : AMDGPUBuiltin<"unsigned int()", [Const]>; + +def __builtin_amdgcn_atomic_inc32 : AMDGPUBuiltin<"uint32_t(uint32_t volatile *, uint32_t, unsigned int, char const *)">; +def __builtin_amdgcn_atomic_inc64 : AMDGPUBuiltin<"uint64_t(uint64_t volatile *, uint64_t, unsigned int, char const *)">; + +def __builtin_amdgcn_atomic_dec32 : AMDGPUBuiltin<"uint32_t(uint32_t volatile *, uint32_t, unsigned int, char const *)">; +def __builtin_amdgcn_atomic_dec64 : AMDGPUBuiltin<"uint64_t(uint64_t volatile *, uint64_t, unsigned int, char const *)">; + +// FIXME: Need to disallow constant address space. +def __builtin_amdgcn_div_scale : AMDGPUBuiltin<"double(double, double, bool, bool *)">; +def __builtin_amdgcn_div_scalef : AMDGPUBuiltin<"float(float, float, bool, bool *)">; +def __builtin_amdgcn_div_fmas : AMDGPUBuiltin<"double(double, double, double, bool)", [Const]>; +def __builtin_amdgcn_div_fmasf : AMDGPUBuiltin<"float(float, float, float, bool)", [Const]>; +def __builtin_amdgcn_div_fixup : AMDGPUBuiltin<"double(double, double, double)", [Const]>; +def __builtin_amdgcn_div_fixupf : AMDGPUBuiltin<"float(float, float, float)", [Const]>; +def __builtin_amdgcn_trig_preop : AMDGPUBuiltin<"double(double, int)", [Const]>; +def __builtin_amdgcn_trig_preopf : AMDGPUBuiltin<"float(float, int)", [Const]>; +def __builtin_amdgcn_rcp : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_rcpf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_sqrt : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_sqrtf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_rsq : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_rsqf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_rsq_clamp : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_rsq_clampf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_sinf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_cosf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_logf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_exp2f : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_log_clampf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_ldexp : AMDGPUBuiltin<"double(double, int)", [Const]>; +def __builtin_amdgcn_ldexpf : AMDGPUBuiltin<"float(float, int)", [Const]>; +def __builtin_amdgcn_frexp_mant : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_frexp_mantf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_frexp_exp : AMDGPUBuiltin<"int(double)", [Const]>; +def __builtin_amdgcn_frexp_expf : AMDGPUBuiltin<"int(float)", [Const]>; +def __builtin_amdgcn_fract : AMDGPUBuiltin<"double(double)", [Const]>; +def __builtin_amdgcn_fractf : AMDGPUBuiltin<"float(float)", [Const]>; +def __builtin_amdgcn_lerp : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const], "lerp-inst">; +def __builtin_amdgcn_class : AMDGPUBuiltin<"bool(double, int)", [Const]>; +def __builtin_amdgcn_classf : AMDGPUBuiltin<"bool(float, int)", [Const]>; +def __builtin_amdgcn_cubeid : AMDGPUBuiltin<"float(float, float, float)", [Const], "cube-insts">; +def __builtin_amdgcn_cubesc : AMDGPUBuiltin<"float(float, float, float)", [Const], "cube-insts">; +def __builtin_amdgcn_cubetc : AMDGPUBuiltin<"float(float, float, float)", [Const], "cube-insts">; +def __builtin_amdgcn_cubema : AMDGPUBuiltin<"float(float, float, float)", [Const], "cube-insts">; +def __builtin_amdgcn_s_sleep : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_s_incperflevel : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_s_decperflevel : AMDGPUBuiltin<"void(_Constant int)">; +def __builtin_amdgcn_s_setprio : AMDGPUBuiltin<"void(_Constant short)">; +def __builtin_amdgcn_ds_swizzle : AMDGPUBuiltin<"int(int, _Constant int)", [Const]>; +def __builtin_amdgcn_ds_permute : AMDGPUBuiltin<"int(int, int)", [Const]>; +def __builtin_amdgcn_ds_bpermute : AMDGPUBuiltin<"int(int, int)", [Const]>; +def __builtin_amdgcn_readfirstlane : AMDGPUBuiltin<"int(int)", [Const]>; +def __builtin_amdgcn_readlane : AMDGPUBuiltin<"int(int, int)", [Const]>; +def __builtin_amdgcn_fmed3f : AMDGPUBuiltin<"float(float, float, float)", [Const]>; +def __builtin_amdgcn_ds_faddf : AMDGPUBuiltin<"float(float address_space<3> *, float, _Constant int, _Constant int, _Constant bool)">; +def __builtin_amdgcn_ds_fminf : AMDGPUBuiltin<"float(float address_space<3> *, float, _Constant int, _Constant int, _Constant bool)">; +def __builtin_amdgcn_ds_fmaxf : AMDGPUBuiltin<"float(float address_space<3> *, float, _Constant int, _Constant int, _Constant bool)">; +def __builtin_amdgcn_ds_append : AMDGPUBuiltin<"int(int address_space<3> *)">; +def __builtin_amdgcn_ds_consume : AMDGPUBuiltin<"int(int address_space<3> *)">; +def __builtin_amdgcn_alignbit : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_alignbyte : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_ubfe : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_sbfe : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_cvt_pkrtz : AMDGPUBuiltin<"_ExtVector<2, __fp16>(float, float)", [Const]>; +def __builtin_amdgcn_cvt_pknorm_i16 : AMDGPUBuiltin<"_ExtVector<2, short>(float, float)", [Const], "cvt-pknorm-vop2-insts">; +def __builtin_amdgcn_cvt_pknorm_u16 : AMDGPUBuiltin<"_ExtVector<2, unsigned short>(float, float)", [Const], "cvt-pknorm-vop2-insts">; +def __builtin_amdgcn_cvt_pk_i16 : AMDGPUBuiltin<"_ExtVector<2, short>(int, int)", [Const]>; +def __builtin_amdgcn_cvt_pk_u16 : AMDGPUBuiltin<"_ExtVector<2, unsigned short>(unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_cvt_pk_u8_f32 : AMDGPUBuiltin<"unsigned int(float, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_cvt_off_f32_i4 : AMDGPUBuiltin<"float(int)", [Const]>; +def __builtin_amdgcn_msad_u8 : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const]>; +def __builtin_amdgcn_sad_u8 : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const], "sad-insts">; +def __builtin_amdgcn_sad_hi_u8 : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const], "sad-insts">; +def __builtin_amdgcn_sad_u16 : AMDGPUBuiltin<"unsigned int(unsigned int, unsigned int, unsigned int)", [Const], "sad-insts">; +def __builtin_amdgcn_qsad_pk_u16_u8 : AMDGPUBuiltin<"uint64_t(uint64_t, unsigned int, uint64_t)", [Const], "qsad-insts">; +def __builtin_amdgcn_mqsad_pk_u16_u8 : AMDGPUBuiltin<"uint64_t(uint64_t, unsigned int, uint64_t)", [Const]>; +def __builtin_amdgcn_mqsad_u32_u8 : AMDGPUBuiltin<"_Vector<4, unsigned int>(uint64_t, unsigned int, _Vector<4, unsigned int>)", [Const]>; ---------------- shiltian wrote:
As a follow-up, I think it's better to use explicit data type, like `uint32_t`, `int32_t`, `uint64_t`, instead of `unsigned`, `int`, etc. We have a mixture of things here, which I'm not sure where they come from. https://github.com/llvm/llvm-project/pull/175873 _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
