提交 f3250097 编写于 作者: T tensor-tang

fix bug and mac compile

上级 bc0df6a9
......@@ -12,9 +12,8 @@
* See the License for the specific language governing permissions and
* limitations under the License. */
// #include <cstring> // for memcpy
// #include <random>
#include <iostream>
#include <random>
#include <string>
#include <vector>
#include "gflags/gflags.h"
......
......@@ -49,6 +49,7 @@ inline typename KernelTuples::func_type GetJitCode(
// pool: (KernelKey(type, place), vector<GenCreatorPtr>)
auto& creator_map = JitCodeCreatorPool().Instance().AllCreators();
auto iter = creator_map.find(kkey);
if (iter != creator_map.end()) {
auto& creators = iter->second;
for (auto& cur : creators) {
auto i = dynamic_cast<const JitCodeCreator<Attr>*>(cur.get());
......@@ -62,6 +63,7 @@ inline typename KernelTuples::func_type GetJitCode(
}
}
}
}
return nullptr;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册