Skip to content

[mypyc] librt base64: use existing SIMD CPU dispatch by customizing build flags #20253

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
JukkaL merged 6 commits into from
Nov 28, 2025
Merged

[mypyc] librt base64: use existing SIMD CPU dispatch by customizing build flags #20253

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
379cd1e
[mypyc] librt base64: use existing SIMD CPU dispatch by customizing b…
mr-c Nov 18, 2025
44e1ebc
[mypyc] librt base64: replace BASE64_WITH_[x86 features] with directl…
mr-c Nov 27, 2025
d0eda17
[mypyc] librt base64: only match against C files when adding the addi…
mr-c Nov 27, 2025
3bc82e6
Merge remote-tracking branch 'upstream/master' into librt_base64_simd…
mr-c Nov 27, 2025
cb0c4ed
[mypyc] librt base64: check for SSE before enabling the rest, to fix …
mr-c Nov 27, 2025
2cf7bff
[mypyc] librt base64: explicitly skip x86-64 CPU features on macos
mr-c Nov 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
[mypyc] librt base64: replace BASE64_WITH_[x86 features] with directl… 
…y using HAVE_[x86 features]
  • Loading branch information
@mr-c
mr-c committed Nov 27, 2025
commit 44e1ebc6f046549d47b8020632eda7c0071d1513
16 changes: 8 additions & 8 deletions mypyc/build_setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,16 +15,16 @@

EXTRA_FLAGS_PER_COMPILER_TYPE_PER_PATH_COMPONENT = {
"unix": {
"base64/arch/ssse3": ["-mssse3", "-DBASE64_WITH_SSSE3"],
"base64/arch/sse41": ["-msse4.1", "-DBASE64_WITH_SSE41"],
"base64/arch/sse42": ["-msse4.2", "-DBASE64_WITH_SSE42"],
"base64/arch/avx2": ["-mavx2", "-DBASE64_WITH_AVX2"],
"base64/arch/avx": ["-mavx", "-DBASE64_WITH_AVX"],
"base64/arch/ssse3": ["-mssse3"],
"base64/arch/sse41": ["-msse4.1"],
"base64/arch/sse42": ["-msse4.2"],
"base64/arch/avx2": ["-mavx2"],
"base64/arch/avx": ["-mavx"],
},
"msvc": {
"base64/arch/sse42": ["/arch:SSE4.2", "/DBASE64_WITH_SSE42"],
"base64/arch/avx2": ["/arch:AVX2", "/DBASE64_WITH_AVX2"],
"base64/arch/avx": ["/arch:AVX", "/DBASE64_WITH_AVX"],
"base64/arch/sse42": ["/arch:SSE4.2"],
"base64/arch/avx2": ["/arch:AVX2"],
"base64/arch/avx": ["/arch:AVX"],
},
}

Expand Down
19 changes: 8 additions & 11 deletions mypyc/lib-rt/base64/config.h
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,17 +1,14 @@
#ifndef BASE64_CONFIG_H
#define BASE64_CONFIG_H

#define HAVE_SSSE3 BASE64_WITH_SSSE3

#define HAVE_SSE41 BASE64_WITH_SSE41

#define HAVE_SSE42 BASE64_WITH_SSE42

#define HAVE_AVX BASE64_WITH_AVX

#define HAVE_AVX2 BASE64_WITH_AVX2

#define HAVE_AVX512 BASE64_WITH_AVX512
#if (defined(__x86_64__) && defined(__LP64__)) || defined(_M_X64)
#define HAVE_SSSE3 1
#define HAVE_SSE41 1
#define HAVE_SSE42 1
#define HAVE_AVX 1
#define HAVE_AVX2 1
#define HAVE_AVX512 0
#endif

#define BASE64_WITH_NEON32 0
#define HAVE_NEON32 BASE64_WITH_NEON32
Expand Down
16 changes: 8 additions & 8 deletions mypyc/lib-rt/setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,16 +27,16 @@

EXTRA_FLAGS_PER_COMPILER_TYPE_PER_PATH_COMPONENT = {
"unix": {
"base64/arch/ssse3": ["-mssse3", "-DBASE64_WITH_SSSE3"],
"base64/arch/sse41": ["-msse4.1", "-DBASE64_WITH_SSE41"],
"base64/arch/sse42": ["-msse4.2", "-DBASE64_WITH_SSE42"],
"base64/arch/avx2": ["-mavx2", "-DBASE64_WITH_AVX2"],
"base64/arch/avx": ["-mavx", "-DBASE64_WITH_AVX"],
"base64/arch/ssse3": ["-mssse3"],
"base64/arch/sse41": ["-msse4.1"],
"base64/arch/sse42": ["-msse4.2"],
"base64/arch/avx2": ["-mavx2"],
"base64/arch/avx": ["-mavx"],
},
"msvc": {
"base64/arch/sse42": ["/arch:SSE4.2", "/DBASE64_WITH_SSE42"],
"base64/arch/avx2": ["/arch:AVX2", "/DBASE64_WITH_AVX2"],
"base64/arch/avx": ["/arch:AVX", "/DBASE64_WITH_AVX"],
"base64/arch/sse42": ["/arch:SSE4.2"],
"base64/arch/avx2": ["/arch:AVX2"],
"base64/arch/avx": ["/arch:AVX"],
},
}

Expand Down