blob: 9bccccf6316dbe533b2b14105ddea1d8e90eb8ea (
plain)
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
|
"""Build definitions for Ruy."""
# 1. Enable -mfpu=neon unconditionally on ARM32. If it turns out that we need to support
# ARM32 without NEON then we'll implement runtime detection and dispatch at that point.
# 2. Explicitly pass -O3 on optimization configs where just "-c opt" means "optimize for code size".
def ruy_copts_base():
return select({
":armeabi-v7a": [
"-mfpu=neon",
],
"//conditions:default": [],
}) + select({
":optimized": ["-O3"],
"//conditions:default": [],
})
# Used for targets that are compiled with extra features that are skipped at runtime if unavailable.
def ruy_copts_skylake():
return []
# Used for targets that are compiled with extra features that are skipped at runtime if unavailable.
def ruy_copts_avx2():
return []
# TODO(b/147376783): SSE 4.2 and AVX-VNNI support is incomplete / placeholder.
# Optimization is not finished. In particular the dimensions of the kernel
# blocks can be changed as desired.
#
# Used for targets that are compiled with extra features that are skipped at runtime if unavailable.
def ruy_copts_sse42():
return []
# TODO(b/147376783): SSE 4.2 and AVX-VNNI support is incomplete / placeholder.
# Optimization is not finished. In particular the dimensions of the kernel
# blocks can be changed as desired.
#
# Used for targets that are compiled with extra features that are skipped at runtime if unavailable.
def ruy_copts_avxvnni():
return []
|