8000 add supported version check. · pytorch/pytorch@bc7aeee · GitHub
[go: up one dir, main page]
Skip to content

Commit bc7aeee

Browse files
xuhancnxuhancn
committed
add supported version check.
1 parent 6d601a9 commit bc7aeee

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

‎aten/src/ATen/mkl/MklAllocationHelp.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
#include <ATen/Version.h>
21
#include <ATen/Config.h>
32

43
#if AT_MKLDNN_ENABLED()
54
#ifdef USE_MIMALLOC_ON_MKL
65
#include <c10/core/impl/alloc_cpu.h>
7-
6+
#include <mkl.h>
7+
#if INTEL_MKL_VERSION > 20230000L
88
/*
99
MKL have a method to register memory allocation APIs via i_malloc.h, High
1010
performance memory allocation APIs will help improve MKL performance.
@@ -26,3 +26,4 @@ bool register_mimalloc_api_to_mkl()
2626
static bool g_b_registered_mkl_alloction = register_mimalloc_api_to_mkl();
2727
#endif
2828
#endif
29+
#endif

0 commit comments

Comments
 (0)
0