-
Notifications
You must be signed in to change notification settings - Fork 78
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
- Loading branch information
Showing
2 changed files
with
86 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,46 @@ | ||
| /**************************************************************************/ | ||
| /* */ | ||
| /* OCaml */ | ||
| /* */ | ||
| /* Max Slater, Jane Street */ | ||
| /* */ | ||
| /* Copyright 2023 Jane Street Group LLC */ | ||
| /* */ | ||
| /* All rights reserved. This file is distributed under the terms of */ | ||
| /* the GNU Lesser General Public License version 2.1, with the */ | ||
| /* special exception on linking described in the file LICENSE. */ | ||
| /* */ | ||
| /**************************************************************************/ | ||
|
|
||
| /* SIMD vector instruction support */ | ||
|
|
||
| #ifndef CAML_SIMD_H | ||
| #define CAML_SIMD_H | ||
|
|
||
| #include "mlvalues.h" | ||
|
|
||
| #if defined(_M_IX86_FP) || defined(__SSE2__) || defined(__SSE3__) || \ | ||
| defined(__SSSE3__) || defined(__SSE4_1__) || defined(__SSE4_2__) | ||
| #define ARCH_SSE2 | ||
| #endif | ||
|
|
||
| #if defined(__AVX__) || defined(__AVX2__) | ||
| #define ARCH_AVX | ||
| #endif | ||
|
|
||
| #ifdef ARCH_SSE2 | ||
| #include <emmintrin.h> | ||
|
|
||
| #define Vec128_val(v) _mm_loadu_ps((const float*)Bp_val(v)) | ||
| #define Vec128_vald(v) _mm_loadu_pd((const double*)Bp_val(v)) | ||
| #define Vec128_vali(v) _mm_loadu_si128((const __m128i*)Bp_val(v)) | ||
| #define Store_vec128_val(v,x) _mm_storeu_ps((float*)Bp_val(v), x) | ||
| #define Store_vec128_vald(v,x) _mm_storeu_pd((double*)Bp_val(v), x) | ||
| #define Store_vec128_vali(v,x) _mm_storeu_si128((__m128i*)Bp_val(v), x) | ||
|
|
||
| CAMLextern value caml_copy_vec128(__m128); | ||
| CAMLextern value caml_copy_vec128i(__m128i); | ||
| CAMLextern value caml_copy_vec128d(__m128d); | ||
| #endif | ||
|
|
||
| #endif /* CAML_SIMD_H */ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| /**************************************************************************/ | ||
| /* */ | ||
| /* OCaml */ | ||
| /* */ | ||
| /* Max Slater, Jane Street */ | ||
| /* */ | ||
| /* Copyright 2023 Jane Street Group LLC */ | ||
| /* */ | ||
| /* All rights reserved. This file is distributed under the terms of */ | ||
| /* the GNU Lesser General Public License version 2.1, with the */ | ||
| /* special exception on linking described in the file LICENSE. */ | ||
| /* */ | ||
| /**************************************************************************/ | ||
|
|
||
| #define CAML_INTERNALS | ||
|
|
||
| #include "caml/alloc.h" | ||
| #include "caml/simd.h" | ||
|
|
||
| #ifdef ARCH_SSE2 | ||
|
|
||
| CAMLexport value caml_copy_vec128(__m128 v) { | ||
| value res = caml_alloc_small(2, Abstract_tag); | ||
| Store_vec128_val(res, v); | ||
| return res; | ||
| } | ||
|
|
||
| CAMLexport value caml_copy_vec128i(__m128i v) { | ||
| value res = caml_alloc_small(2, Abstract_tag); | ||
| Store_vec128_vali(res, v); | ||
| return res; | ||
| } | ||
|
|
||
| CAMLexport value caml_copy_vec128d(__m128d v) { | ||
| value res = caml_alloc_small(2, Abstract_tag); | ||
| Store_vec128_vald(res, v); | ||
| return res; | ||
| } | ||
|
|
||
| #endif |