stream_config.hpp 608 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;
14
15
    int cold_niters_       = 5;
    int nrepeat_           = 50;
ltqin's avatar
ltqin committed
16
17
18

    bool flush_cache   = false;
    int rotating_count = 1;
JD's avatar
JD committed
19
};
letaoqin's avatar
first  
letaoqin committed
20
21
22
23
24
25
26
27
28

struct GemmConfig
{
    int tile_m          = 1;
    int tile_n          = 1;
    int split_k         = 1;
    int stages          = 1;
    std::string op_name = "";
};