mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-11 09:40:51 +00:00
29 lines
608 B
C++
29 lines
608 B
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include <hip/hip_runtime.h>
|
|
#include <hip/hip_fp16.h>
|
|
|
|
struct StreamConfig
|
|
{
|
|
hipStream_t stream_id_ = nullptr;
|
|
bool time_kernel_ = false;
|
|
int log_level_ = 0;
|
|
int cold_niters_ = 5;
|
|
int nrepeat_ = 50;
|
|
|
|
bool flush_cache = false;
|
|
int rotating_count = 1;
|
|
};
|
|
|
|
struct GemmConfig
|
|
{
|
|
int tile_m = 1;
|
|
int tile_n = 1;
|
|
int split_k = 1;
|
|
int stages = 1;
|
|
std::string op_name = "";
|
|
};
|