diff options
author | Daya S Khudia <dskhudia@fb.com> | 2018-12-05 23:37:26 +0300 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-12-05 23:55:02 +0300 |
commit | b1167589a799f83e1eb536d072a79bb5efbdacf5 (patch) | |
tree | c40171e8f2f5fac9c74c6c32a6ae5646fce37c65 /src/OptimizedKernelsAvx2.h | |
parent | bf398ec35647adcc1627fd2ab5999651862b53a3 (diff) |
Removed avx2 code from PackAWithRowOffset.cc (#34)
Summary:
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/34
Removing avx2 intrinsics from PackAWithRowOffset.cc
Reviewed By: jspark1105
Differential Revision: D13281312
fbshipit-source-id: fcaf70fc666674ace26b72b733af03e3b8586e6e
Diffstat (limited to 'src/OptimizedKernelsAvx2.h')
-rw-r--r-- | src/OptimizedKernelsAvx2.h | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/OptimizedKernelsAvx2.h b/src/OptimizedKernelsAvx2.h new file mode 100644 index 0000000..3e6d1d3 --- /dev/null +++ b/src/OptimizedKernelsAvx2.h @@ -0,0 +1,18 @@ +/* + * Copyright (c) Facebook, Inc. and its affiliates. + * All rights reserved. + * This source code is licensed under the BSD-style license found in the + * LICENSE file in the root directory of this source tree. + */ +#pragma once + +#include <cstdint> // for std::int32_t + +namespace fbgemm { + +/** + * @brief Sum a given vector + */ +std::int32_t reduceAvx2(const std::uint8_t* A, int len); + +} // namespace fbgemm |