XeTLA v0.3.6
IntelĀ® Xe Templates for Linear Algebra - API Definition Document
 
Loading...
Searching...
No Matches
common.hpp
Go to the documentation of this file.
1/*******************************************************************************
2* Copyright (c) 2022-2023 Intel Corporation
3*
4* Licensed under the Apache License, Version 2.0 (the "License");
5* you may not use this file except in compliance with the License.
6* You may obtain a copy of the License at
7*
8* http://www.apache.org/licenses/LICENSE-2.0
9*
10* Unless required by applicable law or agreed to in writing, software
11* distributed under the License is distributed on an "AS IS" BASIS,
12* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13* See the License for the specific language governing permissions and
14* limitations under the License.
15*******************************************************************************/
16
19
20#pragma once
21
22#include "common/common.hpp"
23#include "subgroup/subgroup.hpp"
24
25namespace gpu::xetla::group {
26
31template <typename dtype_a_, typename dtype_b_, typename dtype_acc_>
33 using dtype_a = dtype_a_;
34 using dtype_b = dtype_b_;
35 using dtype_acc = dtype_acc_;
36};
37
42template <int k_stride_ = 8, int stages_ = 3, int sync_freq_ = 0>
44 static constexpr int k_stride = k_stride_;
45 static constexpr int stages = stages_;
46 static constexpr int sync_freq = sync_freq_;
47};
48
49} // namespace gpu::xetla::group
Definition limitation.hpp:607
Compute attribute for gemm.
Definition common.hpp:32
dtype_b_ dtype_b
Definition common.hpp:34
dtype_acc_ dtype_acc
Definition common.hpp:35
dtype_a_ dtype_a
Definition common.hpp:33
Fine-tune knobs for gemm.
Definition common.hpp:43
static constexpr int stages
Definition common.hpp:45
static constexpr int sync_freq
Definition common.hpp:46
static constexpr int k_stride
Definition common.hpp:44