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

JD's avatar
JD committed
4
5
6
7
8
9
10
11
12
#pragma once

#include <hip/hip_runtime.h>
#include <hip/hip_fp16.h>

struct StreamConfig
{
    hipStream_t stream_id_ = nullptr;
    bool time_kernel_      = false;
13
    int log_level_         = 0;
Jing Zhang's avatar
Jing Zhang committed
14
15
    int cold_niters_       = 1;
    int nrepeat_           = 10;
JD's avatar
JD committed
16
};