Skip to content

Commit

Permalink
Generalize GEMM Benchmark (pytorch#447)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#447

- Templetize the benchmark function to support multiple types
- add arguments to run different FP types

Reviewed By: dskhudia

Differential Revision: D22152357

fbshipit-source-id: fa9f00b3c4e7324df99abb3ee070355b649be2ee
  • Loading branch information
efiks authored and facebook-github-bot committed Nov 9, 2020
1 parent c438e6e commit eb55572
Show file tree
Hide file tree
Showing 3 changed files with 355 additions and 342 deletions.
10 changes: 10 additions & 0 deletions bench/BenchUtils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -114,4 +114,14 @@ bool parseArgumentBool(
}
return def_val;
}

#if defined(USE_MKL)
void test_xerbla(char* srname, const int* info, int) {
// srname - name of the function that called xerbla
// info - position of the invalid parameter in the parameter list
// len - length of the name in bytes
printf("\nXERBLA(MKL Error) is called :%s: %d\n", srname, *info);
}
#endif

} // namespace fbgemm
Loading

0 comments on commit eb55572

Please sign in to comment.