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 @@
#include "ngraph/runtime/backend.hpp"
#include "ngraph/runtime/cpu/cpu_external_function.hpp"
#include "ngraph/runtime/cpu/cpu_tensor_view.hpp"
#include "ngraph/runtime/reference/reduce_window.hpp"
......@@ -45,9 +46,9 @@ namespace ngraph
auto reducer = [&](ElementType a, ElementType b) {
TensorViewPtrs inputs, outputs;
ElementType p __attribute__((aligned(64))) = a;
ElementType q __attribute__((aligned(64))) = b;
ElementType r __attribute__((aligned(64)));
ElementType p __attribute__((aligned(CPUTensorView::BufferAlignment))) = a;
ElementType q __attribute__((aligned(CPUTensorView::BufferAlignment))) = b;
ElementType r __attribute__((aligned(CPUTensorView::BufferAlignment)));
inputs.emplace_back(backend->create_tensor(
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