File: host_function_worker_counting_semaphore.h

package info (click to toggle)
intel-compute-runtime 25.48.36300.8-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 80,652 kB
  • sloc: cpp: 939,022; lisp: 2,090; sh: 722; makefile: 162; python: 21
file content (38 lines) | stat: -rw-r--r-- 1,014 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
/*
 * Copyright (C) 2025 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#pragma once

#include "shared/source/command_stream/host_function_worker_interface.h"

#include <mutex>
#include <semaphore>
#include <thread>

namespace NEO {

class HostFunctionWorkerCountingSemaphore final : public IHostFunctionWorker {
  public:
    HostFunctionWorkerCountingSemaphore(bool skipHostFunctionExecution,
                                        const std::function<void(GraphicsAllocation &)> &downloadAllocationImpl,
                                        GraphicsAllocation *allocation,
                                        HostFunctionData *data);
    ~HostFunctionWorkerCountingSemaphore() override;

    void start() override;
    void finish() override;
    void submit() noexcept override;

  private:
    void workerLoop(std::stop_token st) noexcept;

    std::counting_semaphore<> semaphore{0};
};

static_assert(NonCopyableAndNonMovable<HostFunctionWorkerCountingSemaphore>);

} // namespace NEO