[libclc] Add ctz built-in implementation to clc and generic (#135309)

This commit is contained in:
Wenju He
2025-04-15 14:23:25 +00:00
committed by GitHub
parent 7f2587a239
commit 552902455c
7 changed files with 109 additions and 0 deletions

View File

@@ -0,0 +1,20 @@
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef __CLC_INTEGER_CLC_CTZ_H__
#define __CLC_INTEGER_CLC_CTZ_H__
#define __CLC_FUNCTION __clc_ctz
#define __CLC_BODY <clc/shared/unary_decl.inc>
#include <clc/integer/gentype.inc>
#undef __CLC_BODY
#undef __CLC_FUNCTION
#endif // __CLC_INTEGER_CLC_CTZ_H__

View File

@@ -7,6 +7,7 @@ integer/clc_abs.cl
integer/clc_abs_diff.cl
integer/clc_add_sat.cl
integer/clc_clz.cl
integer/clc_ctz.cl
integer/clc_hadd.cl
integer/clc_mad24.cl
integer/clc_mad_sat.cl

View File

@@ -0,0 +1,48 @@
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#include <clc/clcmacro.h>
#include <clc/integer/clc_ctz.h>
#include <clc/internal/clc.h>
_CLC_OVERLOAD _CLC_DEF char __clc_ctz(char x) {
return __clc_ctz(__clc_as_uchar(x));
}
_CLC_OVERLOAD _CLC_DEF uchar __clc_ctz(uchar x) { return __builtin_ctzg(x, 8); }
_CLC_OVERLOAD _CLC_DEF short __clc_ctz(short x) {
return __clc_ctz(__clc_as_ushort(x));
}
_CLC_OVERLOAD _CLC_DEF ushort __clc_ctz(ushort x) {
return __builtin_ctzg(x, 16);
}
_CLC_OVERLOAD _CLC_DEF int __clc_ctz(int x) {
return __clc_ctz(__clc_as_uint(x));
}
_CLC_OVERLOAD _CLC_DEF uint __clc_ctz(uint x) { return __builtin_ctzg(x, 32); }
_CLC_OVERLOAD _CLC_DEF long __clc_ctz(long x) {
return __clc_ctz(__clc_as_ulong(x));
}
_CLC_OVERLOAD _CLC_DEF ulong __clc_ctz(ulong x) {
return __builtin_ctzg(x, 64);
}
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, char, __clc_ctz, char)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, uchar, __clc_ctz, uchar)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, short, __clc_ctz, short)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, ushort, __clc_ctz, ushort)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, int, __clc_ctz, int)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, uint, __clc_ctz, uint)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, long, __clc_ctz, long)
_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, ulong, __clc_ctz, ulong)

View File

@@ -153,6 +153,7 @@
#include <clc/integer/abs_diff.h>
#include <clc/integer/add_sat.h>
#include <clc/integer/clz.h>
#include <clc/integer/ctz.h>
#include <clc/integer/hadd.h>
#include <clc/integer/mad24.h>
#include <clc/integer/mad_hi.h>

View File

@@ -0,0 +1,19 @@
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#if __OPENCL_C_VERSION__ >= CL_VERSION_2_0
#define __CLC_FUNCTION ctz
#define __CLC_BODY <clc/shared/unary_decl.inc>
#include <clc/integer/gentype.inc>
#undef __CLC_BODY
#undef __CLC_FUNCTION
#endif // __OPENCL_C_VERSION__ >= CL_VERSION_2_0

View File

@@ -44,6 +44,7 @@ integer/abs.cl
integer/abs_diff.cl
integer/add_sat.cl
integer/clz.cl
integer/ctz.cl
integer/hadd.cl
integer/mad24.cl
integer/mad_hi.cl

View File

@@ -0,0 +1,19 @@
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#if __OPENCL_C_VERSION__ >= CL_VERSION_2_0
#include <clc/clc.h>
#include <clc/integer/clc_ctz.h>
#define FUNCTION ctz
#define __CLC_BODY <clc/shared/unary_def.inc>
#include <clc/integer/gentype.inc>
#endif // __OPENCL_C_VERSION__ >= CL_VERSION_2_0