commit 5b5f8d67ea0ec149f3dd32f20b74b56396c26b26
author Anish Tondwalkar <atondwal@google.com> Wed Nov 04 11:34:42 2020 -0800
committer Copybara-Service <iree-copybara-bot@google.com> Wed Nov 04 11:35:41 2020 -0800
tree 321d8ec2b0654de07ecb4b622930bc04c02f8515
parent 583c5ea8dc4830fb12ea6435f5a52fe72eff1ad1

fix type of makeInitializedUnrankedDescriptor

PiperOrigin-RevId: 340694521

experimental/ModelBuilder/MemRefUtils.h

diff --git a/experimental/ModelBuilder/MemRefUtils.h b/experimental/ModelBuilder/MemRefUtils.h
index 9391569..317a517 100644
--- a/experimental/ModelBuilder/MemRefUtils.h
+++ b/experimental/ModelBuilder/MemRefUtils.h
@@ -211,7 +211,7 @@
 // of type PointwiseInitializer. Can optionally take specific `alloc` and `free`
 // functions.
 template <typename T, int N, typename FreeFunType = decltype(&::free)>
-std::unique_ptr<::UnrankedMemRefType<float>, FreeFunType>
+std::unique_ptr<::UnrankedMemRefType<T>, FreeFunType>
 makeInitializedUnrankedDescriptor(
     const std::array<int64_t, N> &shape, LinearInitializer<T> init,
     llvm::Optional<uint64_t> alignment = llvm::Optional<uint64_t>(),
@@ -222,7 +222,7 @@
   auto *data = static_cast<T *>(allocated.first);
   auto *alignedData = static_cast<T *>(allocated.second);
   for (unsigned i = 0; i < nElements; ++i) init(i, alignedData);
-  return std::unique_ptr<::UnrankedMemRefType<float>, FreeFunType>(
+  return std::unique_ptr<::UnrankedMemRefType<T>, FreeFunType>(
       detail::allocUnrankedDescriptor<T, N>(data, alignedData, shape, alloc),
       freeFun);
 }