templating folly::fibers::await by Baton

[folly.git] / folly / fibers / FiberManager-inl.h

diff --git a/folly/fibers/FiberManager-inl.h b/folly/fibers/FiberManager-inl.h
index 0e14525dec1b9c0b50e8d0634b6703dbbf31e708..003da1bdb1186ff0adc95725c082a2e0874a6f4b 100644 (file)
--- a/folly/fibers/FiberManager-inl.h
+++ b/folly/fibers/FiberManager-inl.h
@@ -545,8 +545,9 @@ FiberManager::FiberManager(
 template <typename F>
 typename FirstArgOf<F>::type::value_type inline await(F&& func) {
   typedef typename FirstArgOf<F>::type::value_type Result;
+  typedef typename FirstArgOf<F>::type::baton_type BatonT;
 
-  return Promise<Result>::await(std::forward<F>(func));
+  return Promise<Result, BatonT>::await(std::forward<F>(func));
 }
 }
 }