fe1bce6717
PR #42339: [INTEL MKL] MKL DNN v0.x cleanup - Reset MKL build config and remove DNN v0.x related macros Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/42339 This is the first PR for MKL DNN v0.x clean - for Tensorflow 2.4, only MKL DNN v1.x will be supported (1) Reset MKL DNN related build config (mostly in third_part/mkl or mkl_dnn folder) (2) Remove MKL DNN v0.x related macros in core/util/mkl_types.c (3) Minor code style fix in one MKL kernel op. There will be a sequence of PR's which will clean up all related MKL kernels ops. PiperOrigin-RevId: 331205640 Change-Id: I2c3c3671fe8906a5dd18c6b017d05fc69a1e3f59
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
def if_mkl_open_source_only(if_true, if_false = []):
|
|
"""Returns `if_true` if MKL-DNN v0.x is used.
|
|
|
|
Shorthand for select()'ing on whether we're building with
|
|
MKL-DNN v0.x open source library only, without depending on MKL binary form.
|
|
|
|
Returns a select statement which evaluates to if_true if we're building
|
|
with MKL-DNN v0.x open source library only. Otherwise, the select statement
|
|
evaluates to if_false.
|
|
|
|
"""
|
|
return select({
|
|
"@org_tensorflow//third_party/mkl_dnn:build_with_mkl_opensource": if_true,
|
|
"//conditions:default": if_false,
|
|
})
|
|
|
|
def if_mkldnn_threadpool(if_true, if_false = []):
|
|
"""Returns `if_true` if MKL-DNN v1.x is used.
|
|
|
|
Shorthand for select()'ing on whether we're building with
|
|
MKL-DNN v1.x open source library only with user specified threadpool, without depending on MKL binary form.
|
|
|
|
Returns a select statement which evaluates to if_true if we're building
|
|
with MKL-DNN v1.x open source library only with user specified threadpool. Otherwise, the
|
|
select statement evaluates to if_false.
|
|
|
|
"""
|
|
return select({
|
|
"@org_tensorflow//third_party/mkl_dnn:build_with_mkldnn_threadpool": if_true,
|
|
"//conditions:default": if_false,
|
|
})
|