Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/FBGEMM.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDaya S Khudia <dskhudia@fb.com>2018-11-20 04:05:07 +0300
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>2018-11-20 04:09:44 +0300
commit3b7936e008865c6c59773664c20dd98c73b55c82 (patch)
treec596a28d4be27478b0133954f900e8bacb398553 /include/fbgemm/Fbgemm.h
parent6d00b52ee80787065561ac16942e55042ed6c760 (diff)
A function to check if we are running on a fbgemm supported cpu (#13)
Summary: Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/13 See title Reviewed By: jianyuh Differential Revision: D13131301 fbshipit-source-id: 2dafdf0fe3dfd26f1b944d550d6cce29f3653a74
Diffstat (limited to 'include/fbgemm/Fbgemm.h')
-rw-r--r--include/fbgemm/Fbgemm.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/include/fbgemm/Fbgemm.h b/include/fbgemm/Fbgemm.h
index d05e099..9cb87ea 100644
--- a/include/fbgemm/Fbgemm.h
+++ b/include/fbgemm/Fbgemm.h
@@ -1019,4 +1019,9 @@ static void* fbgemmAlignedAlloc(size_t __align, size_t __size) {
return aligned_mem;
}
+/**
+ * @brief Are we running on a fbgemm supported cpu?
+ */
+bool fbgemmSupportedCPU();
+
} // namespace fbgemm