mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-30 03:37:38 +00:00
41 lines
1.1 KiB
C
41 lines
1.1 KiB
C
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#if defined(KERNEL_A)
|
|
#define PADDING_K_FIRST
|
|
#define USING_MFMA_32x32x_8x2
|
|
#elif defined(KERNEL_B)
|
|
#define PADDING_K_FIRST
|
|
#define USING_MFMA_16x16x16
|
|
#elif defined(KERNEL_C)
|
|
#define PADDING_K_FIRST
|
|
#define USING_MFMA_16x16x_16x2
|
|
#elif defined(KERNEL_D)
|
|
#define USING_MFMA_16x16x_16x2
|
|
#define USING_XOR_BASED_BANK_CONFLICT_FREE
|
|
#elif defined(KERNEL_E)
|
|
#define USING_MFMA_16x16x_16x2
|
|
#define USING_XOR_BASED_BANK_CONFLICT_FREE
|
|
#define ADJUST_BLOCK_TILE_SHAPE
|
|
#elif defined(KERNEL_F)
|
|
#define USING_MFMA_16x16x_16x2
|
|
#define USING_XOR_BASED_BANK_CONFLICT_FREE
|
|
#define ADJUST_BLOCK_TILE_SHAPE
|
|
#define ENABLE_PREFETCH
|
|
#elif defined(KERNEL_G)
|
|
#define USING_MFMA_16x16x_16x2
|
|
#define USING_XOR_BASED_BANK_CONFLICT_FREE
|
|
#define ADJUST_BLOCK_TILE_SHAPE
|
|
#define ENABLE_PREFETCH
|
|
#define ENABLE_INSTRUCTION_SCH
|
|
#elif defined(KERNEL_H)
|
|
#define USING_MFMA_16x16x_16x2
|
|
#define USING_XOR_BASED_BANK_CONFLICT_FREE
|
|
#define ADJUST_BLOCK_TILE_SHAPE
|
|
#define ENABLE_PREFETCH
|
|
#define ENABLE_INSTRUCTION_SCH
|
|
#define ENABLE_CACHE_AWARE_WG_SCH
|
|
#else
|
|
#define NAIVE_IMPLEMENTATION
|
|
#endif
|