bit_mask_manipulation.hpp

Go to the documentation of this file.

/*******************************************************************************
* Copyright (c) 2022-2023 Intel Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*******************************************************************************/
 
 
#pragma once
 
#include "common/core/base_types.hpp"
#include "common/core/common.hpp"
 
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wunused-parameter"
namespace gpu::xetla {
 
 
template <typename T0, typename T1, int SZ, typename U,
        class Sat = xetla_saturation_off_tag>
__XETLA_API xetla_vector<T0, SZ> xetla_shl(
        xetla_vector<T1, SZ> src0, U src1, Sat sat = {}) {
    return __ESIMD_ENS::shl<T0, T1, SZ, U, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, typename T2,
        class Sat = xetla_saturation_off_tag>
typename std::remove_const<T0>::type xetla_shl(T1 src0, T2 src1, Sat sat = {}) {
    return __ESIMD_ENS::shl<T0, T1, T2, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, int SZ, typename U,
        class Sat = xetla_saturation_off_tag>
__XETLA_API xetla_vector<T0, SZ> xetla_shr(
        xetla_vector<T1, SZ> src0, U src1, Sat sat = {}) {
    return __ESIMD_ENS::shr<T0, T1, SZ, U, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, typename T2,
        class Sat = xetla_saturation_off_tag>
__XETLA_API typename std::remove_const<T0>::type xetla_shr(
        T1 src0, T2 src1, Sat sat = {}) {
    return __ESIMD_ENS::shr<T0, T1, T2, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, int SZ>
__XETLA_API xetla_vector<T0, SZ> xetla_rol(
        xetla_vector<T1, SZ> src0, xetla_vector<T1, SZ> src1) {
    return __ESIMD_ENS::rol<T0, T1, SZ>(src0, src1);
}
 
template <typename T0, typename T1, int SZ, typename U>
__XETLA_API std::enable_if_t<std::is_integral<T0>::value
                && std::is_integral<T1>::value && std::is_integral<U>::value
                && is_xetla_scalar<U>::value,
        xetla_vector<T0, SZ>>
xetla_rol(xetla_vector<T1, SZ> src0, U src1) {
    return __ESIMD_ENS::rol<T0, T1, SZ, U>(src0, src1);
}
 
template <typename T0, typename T1, typename T2>
__XETLA_API std::enable_if_t<std::is_integral<T0>::value
                && std::is_integral<T1>::value && std::is_integral<T2>::value,
        remove_const_t<T0>>
xetla_rol(T1 src0, T2 src1) {
    return __ESIMD_ENS::rol<T0, T1, T2>(src0, src1);
}
 
template <typename T0, typename T1, int SZ>
__XETLA_API xetla_vector<T0, SZ> xetla_ror(
        xetla_vector<T1, SZ> src0, xetla_vector<T1, SZ> src1) {
    return __ESIMD_ENS::ror<T0, T1, SZ>(src0, src1);
}
 
template <typename T0, typename T1, int SZ, typename U>
__XETLA_API std::enable_if_t<std::is_integral<T0>::value
                && std::is_integral<T1>::value && std::is_integral<U>::value
                && is_xetla_scalar<U>::value,
        xetla_vector<T0, SZ>>
xetla_ror(xetla_vector<T1, SZ> src0, U src1) {
    return __ESIMD_ENS::ror<T0, T1, SZ, U>(src0, src1);
}
 
template <typename T0, typename T1, typename T2>
__XETLA_API std::enable_if_t<std::is_integral<T0>::value
                && std::is_integral<T1>::value && std::is_integral<T2>::value,
        remove_const_t<T0>>
xetla_ror(T1 src0, T2 src1) {
    return __ESIMD_ENS::ror<T0, T1, T2>(src0, src1);
}
 
template <typename T0, typename T1, int SZ, typename U,
        class Sat = xetla_saturation_off_tag>
__XETLA_API xetla_vector<T0, SZ> xetla_lsr(
        xetla_vector<T1, SZ> src0, U src1, Sat sat = {}) {
    return __ESIMD_ENS::lsr<T0, T1, SZ, U, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, typename T2,
        class Sat = xetla_saturation_off_tag>
__XETLA_API typename std::remove_const<T0>::type xetla_lsr(
        T1 src0, T2 src1, Sat sat = {}) {
    return __ESIMD_ENS::lsr<T0, T1, T2, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, int SZ, typename U,
        class Sat = xetla_saturation_off_tag>
__XETLA_API xetla_vector<T0, SZ> xetla_asr(
        xetla_vector<T1, SZ> src0, U src1, Sat sat = {}) {
    return __ESIMD_ENS::asr<T0, T1, SZ, U, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <typename T0, typename T1, typename T2,
        class Sat = xetla_saturation_off_tag>
__XETLA_API typename std::remove_const<T0>::type xetla_asr(
        T1 src0, T2 src1, Sat sat = {}) {
    return __ESIMD_ENS::asr<T0, T1, T2, typename Sat::sat_tag>(
            src0, src1, Sat::value);
}
 
template <int N>
__XETLA_API uint32_t xetla_pack_mask(xetla_mask<N> src0) {
    return __ESIMD_NS::pack_mask<N>(src0);
}
 
template <int N>
__XETLA_API xetla_mask<N> xetla_unpack_mask(uint32_t src0) {
    return __ESIMD_NS::unpack_mask<N>(src0);
}
 
 
} // namespace gpu::xetla
#pragma clang diagnostic pop