diff options
author | jokeren <robinho364@gmail.com> | 2016-12-17 09:00:15 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-02-23 13:50:34 +0300 |
commit | 8137de1b1258bd38e5bb582b41e0fed02cbfbb08 (patch) | |
tree | e1d230e6ef96fbb5c0cdffed72324d5a1cdbfb21 | |
parent | 084a856cbe3f8f97c48e9679826ee2dce814e75f (diff) |
Add THVector Fill
-rw-r--r-- | lib/TH/generic/THVectorDispatch.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/TH/generic/THVectorDispatch.c b/lib/TH/generic/THVectorDispatch.c index 7537f66..9563f48 100644 --- a/lib/TH/generic/THVectorDispatch.c +++ b/lib/TH/generic/THVectorDispatch.c @@ -26,6 +26,12 @@ static FunctionDescription THVector_(fill_DISPATCHTABLE)[] = { #endif #endif + #if defined(USE_AVX) + #if defined(TH_REAL_IS_DOUBLE) || defined(TH_REAL_IS_FLOAT) + FUNCTION_IMPL(THVector_(fill_AVX), SIMDExtension_AVX), + #endif + #endif + #if defined(USE_SSE2) || defined(USE_SSE3) || defined(USE_SSSE3) \ || defined(USE_SSE4_1) || defined(USE_SSE4_2) #if defined(TH_REAL_IS_DOUBLE) || defined(TH_REAL_IS_FLOAT) |