XeTLA v0.3.6
IntelĀ® Xe Templates for Linear Algebra - API Definition Document
 
Loading...
Searching...
No Matches
dispatch_policy.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
23
24namespace gpu::xetla::kernel {
25
28
36template <typename group_swizzle_policy_, int num_global_kslicing_ = 1,
37 int num_local_kslicing_ = 1>
39 using group_swizzle_policy = group_swizzle_policy_;
40 static constexpr int num_global_kslicing = num_global_kslicing_;
41 static constexpr int num_local_kslicing = num_local_kslicing_;
42 static constexpr gpu_arch arch_tag = group_swizzle_policy::arch_tag;
43};
44
46
47} // namespace gpu::xetla::kernel
C++ API.
Definition limitation.hpp:734
gpu_arch
Definition common.hpp:73
4bit kslicing GEMM implementation.
Definition dispatch_policy.hpp:38
static constexpr int num_global_kslicing
Definition dispatch_policy.hpp:40
group_swizzle_policy_ group_swizzle_policy
Definition dispatch_policy.hpp:39
static constexpr gpu_arch arch_tag
Definition dispatch_policy.hpp:42
static constexpr int num_local_kslicing
Definition dispatch_policy.hpp:41