ytensor/ytensor.hpp at main · SnifferCaptain/ytensor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#pragma once
/***************
* @file: ytensor.hpp
* @brief: 所有include文件的总入口。
* @author: SnifferCaptain
* @date: 2026-1-10
* @version 0.3
* @email: 3586554865@qq.com
***************/

#include "include/ytensor_concepts.hpp"
#include "include/ytensor_infos.hpp"
#include "include/ytensor_types.hpp"
#include "include/kernel/parallel_for.hpp"
#include "include/kernel/type_dispatch.hpp"

// AVX2 kernel
#include "include/kernel/avx2/hgemm.hpp"
#include "include/kernel/avx2/sgemm.hpp"
#include "include/kernel/avx2/sgemv.hpp"

// Backend switch: define YT_USE_LIB to use precompiled runtime backend.

/////////// ytensor class def ////////////
#include "include/ytensor_base.hpp"
#include "include/ytensor_core.hpp"

//////////// external /////////////
#include "include/ytensor_function.hpp"
#include "include/ytensor_io.hpp"

//////////// implementation /////////////
#include "src/ytensor_base_templates.inl"
#include "src/ytensor_io_templates.inl"

#if !YT_USE_LIB || defined(YT_LIBRARY_IMPLEMENTATION)
#include "src/ytensor_base.inl"
#include "src/ytensor_io.inl"
#endif

#if !YT_USE_LIB || defined(YT_LIBRARY_IMPLEMENTATION)
#include "src/ytensor_base_math.inl"
#endif

#include "src/ytensor_core.inl"
#include "src/ytensor_math.inl"

#include "src/ytensor_function.inl"

// In YT_USE_LIB consumer mode, suppress repeated builtin template instantiation.
#if YT_USE_LIB && !defined(YT_LIBRARY_IMPLEMENTATION)
#include "include/ytensor_extern_templates.hpp"
#endif