stream_config.hpp 328 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.

JD's avatar
JD committed
4
5
#pragma once

6
#ifndef CK_NOGPU
JD's avatar
JD committed
7
8
#include <hip/hip_runtime.h>
#include <hip/hip_fp16.h>
9
#endif
JD's avatar
JD committed
10
11
12

struct StreamConfig
{
13
#ifndef CK_NOGPU
JD's avatar
JD committed
14
    hipStream_t stream_id_ = nullptr;
15
#endif
JD's avatar
JD committed
16
17
    bool time_kernel_      = false;
};