InfiniTensor/test/nnet/readlog.cc

18 lines
493 B
C++
Raw Normal View History

2022-08-08 16:02:07 +08:00
#include "nnet/Visitor/FullPrinterVisitor.h"
#include "nnet/Visitor/HashVisitor.h"
#include "nnet/Visitor/Serializer.h"
#include "nnet/expr.h"
using namespace nnet;
using namespace std;
int main(int argc, char *argv[]) {
if (argc != 2) {
printf("Usage: %s <log>\n", argv[0]);
return 1;
}
auto expr = Serializer().fromFile(argv[1]);
2022-08-08 16:02:07 +08:00
cout << FullPrinterVisitor().print(expr);
cout << endl << "Hash = " << HashVisitor().getHash(expr) << endl;
return 0;
}