mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-01 12:11:19 +00:00
27e0a34e0f
[CK] CK Tile grouped convolution direct load ## Motivation CK Tile grouped convolution forward direct load support. ## Technical Details Basic pipeline for direct load and new instances for forward for v1 and v4 pipelines. ## Test Plan test_grouped_convnd_fwd_tile ## Test Result CI pending ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests. AICK-130
23 lines
349 B
C++
23 lines
349 B
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
namespace ck_tile {
|
|
|
|
enum struct GemmPipeline
|
|
{
|
|
COMPUTE_ASYNC,
|
|
COMPUTE_V3,
|
|
COMPUTE_V4,
|
|
COMPUTE_V5,
|
|
COMPUTE_V6,
|
|
MEMORY,
|
|
BASIC_V1,
|
|
BASIC_V2,
|
|
PRESHUFFLE_V2,
|
|
BASIC_ASYNC_V1
|
|
};
|
|
|
|
} // namespace ck_tile
|