Commit 85fad484 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon Committed by Scott Cyphers

Replace hardcoded alignment value (#1366)

parent 6b3a668c
...@@ -18,6 +18,7 @@ ...@@ -18,6 +18,7 @@
#include "ngraph/runtime/backend.hpp" #include "ngraph/runtime/backend.hpp"
#include "ngraph/runtime/cpu/cpu_external_function.hpp" #include "ngraph/runtime/cpu/cpu_external_function.hpp"
#include "ngraph/runtime/cpu/cpu_tensor_view.hpp"
#include "ngraph/runtime/reference/reduce_window.hpp" #include "ngraph/runtime/reference/reduce_window.hpp"
...@@ -45,9 +46,9 @@ namespace ngraph ...@@ -45,9 +46,9 @@ namespace ngraph
auto reducer = [&](ElementType a, ElementType b) { auto reducer = [&](ElementType a, ElementType b) {
TensorViewPtrs inputs, outputs; TensorViewPtrs inputs, outputs;
ElementType p __attribute__((aligned(64))) = a; ElementType p __attribute__((aligned(CPUTensorView::BufferAlignment))) = a;
ElementType q __attribute__((aligned(64))) = b; ElementType q __attribute__((aligned(CPUTensorView::BufferAlignment))) = b;
ElementType r __attribute__((aligned(64))); ElementType r __attribute__((aligned(CPUTensorView::BufferAlignment)));
inputs.emplace_back(backend->create_tensor( inputs.emplace_back(backend->create_tensor(
ngraph::element::from<ElementType>(), Shape{}, &p)); ngraph::element::from<ElementType>(), Shape{}, &p));
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment