stream_config.hpp 217 Bytes
Newer Older
JD's avatar
JD committed
1
2
#pragma once

3
#ifndef CK_NOGPU
JD's avatar
JD committed
4
5
#include <hip/hip_runtime.h>
#include <hip/hip_fp16.h>
6
#endif
JD's avatar
JD committed
7
8
9

struct StreamConfig
{
10
#ifndef CK_NOGPU
JD's avatar
JD committed
11
    hipStream_t stream_id_ = nullptr;
12
#endif
JD's avatar
JD committed
13
14
    bool time_kernel_      = false;
};