build with deeprec 2208 in cu114

Author: Orion34-lanbo

scripts/python/common_setup.py

@@ -366,14 +366,15 @@ def update_cpu_specific_setting(args):
 
 def get_tf_info(python_executable):
     output = subprocess.check_output(
-        '{} -c "import tensorflow as tf; print(tf.__version__); print(\'\\n\'.join(tf.sysconfig.get_compile_flags())); print(\'\\n\'.join(tf.sysconfig.get_link_flags()))"'.format(
+        '{} -c "import tensorflow as tf; print(tf.__version__); print(tf.__git_version__); print(\'\\n\'.join(tf.sysconfig.get_compile_flags())); print(\'\\n\'.join(tf.sysconfig.get_link_flags()))"'.format(
             python_executable
         ),
         shell=True,
     ).decode()
     lines = output.split("\n")
     major, minor, _ = lines[0].split(".")  # lines[0] is version like 1.15.0
     is_pai = "PAI" in lines[0]
+    is_deeprec = 'deeprec' in lines[1]
     header_dir, lib_dir, lib_name, cxx11_abi = '', '', '', ''
     for line in lines[1:]:
         if line.startswith("-I"):
@@ -402,7 +403,7 @@ def get_tf_info(python_executable):
             raise Exception("Can not find tensorflow's built-in pb version!")
     else:
         raise Exception("Can not find {PB_HEADER_FILE} in tf's include dir!")
-    return major, minor, is_pai, header_dir, lib_dir, lib_name, cxx11_abi, tf_pb_version
+    return major, minor, is_pai, header_dir, lib_dir, lib_name, cxx11_abi, tf_pb_version, is_deeprec
 
 def deduce_cuda_info():
     """Deduce cuda major and minor version and cuda directory."""

scripts/python/tao_build.py

@@ -231,9 +231,13 @@ def _write(line, cmd="build"):
             tf_lib_name,
             tf_cxx11_abi,
             tf_pb_version,
+            is_deeprec
         ) = get_tf_info(python_bin)
         _opt("cxxopt", f"-D_GLIBCXX_USE_CXX11_ABI={tf_cxx11_abi}")
         _opt("host_cxxopt", f"-D_GLIBCXX_USE_CXX11_ABI={tf_cxx11_abi}")
+        # quick solution, a little hacky
+        _opt("cxxopt", f"-DTF_IS_DEEPREC={is_deeprec}")
+        _opt("host_cxxopt", f"-DTF_IS_DEEPREC={is_deeprec}")
         _action_env("BLADE_WITH_TF", "1")
         _action_env("IF_CXX11_ABI", int(tf_cxx11_abi))
         _action_env("TF_IS_PAI", int(is_pai))

tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc

@@ -1327,6 +1327,9 @@ Status RunCudnnConvolution<int8_t>(
     se::DeviceMemoryBase& result_buffer,
     se::ScratchAllocator* scratch_allocator, se::Stream* stream,
     se::dnn::ProfileResult* profile_result) {
+#if defined (TF_IS_DEEPREC)
+  return errors::Internal("Not supported for DeepREC yet");
+#else
   ConvolutionKind kind = params.kind;
   DeviceMemory<int8_t> input_buf(operand_buffers[0]);
   DeviceMemory<int8_t> filter_buf(operand_buffers[1]);
@@ -1401,6 +1404,7 @@ Status RunCudnnConvolution<int8_t>(
     return errors::Internal("Unable to launch convolution");
   }
   return Status::OK();
+#endif // TF_IS_DEEPREC
 }
 
 template <typename T>