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

fix bug and mac compile

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