raw_send_nbarrier.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/utils/common.hpp"
 
namespace gpu::xetla {
 
 
enum class nbarrier_role : uint8_t {
    producer_consumer = 0,
    producer = 1,
    consumer = 2
};
 
template <uint8_t num_producers = 1, uint8_t num_consumers = 1,
        gpu_arch arch_tag = gpu_arch::Xe>
struct xetla_nbarrier_t {
    xetla_vector<uint32_t, 16> nbar;
    uint32_t barrier_id;
 
    __XETLA_API void init_nbarrier(uint8_t nbarrier_id,
            nbarrier_role role = nbarrier_role::producer_consumer) {
        nbar[2] = (uint32_t)nbarrier_id | uint32_t((uint8_t)role << 14)
                | uint32_t(num_producers << 16) | uint32_t(num_consumers << 24);
        barrier_id = nbarrier_id;
    }
 
    __XETLA_API void arrive() {
        constexpr uint32_t sfid = 0x3;
        constexpr uint32_t exDesc = sfid;
        constexpr uint32_t msg_desc = 0x2000004;
        constexpr uint32_t execSize = 0;
 
        xetla_raw_send<uint32_t, 16, execSize, sfid, 1>(nbar, exDesc, msg_desc);
    }
 
    __XETLA_API void wait() { named_barrier_wait(barrier_id); }
 
    __XETLA_API void arrive_wait() {
        arrive();
        wait();
    }
};
 
 
} // namespace gpu::xetla