stream_config.hpp 455 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
#pragma once

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

struct StreamConfig
{
11
12
13
14
15
16
    hipStream_t stream_id_    = nullptr;
    bool time_kernel_         = false;
    int log_level_            = 0;
    mutable int cold_niters_  = 5;
    mutable int nrepeat_      = 50;
    mutable int time_limit_ms = 1000; // for timing MI300 runs
JD's avatar
JD committed
17
};