Merge pull request #41 from Archana-Shinde1/add-filter

Added filter

Author: Archana-Shinde1

buildscripts/set_tf_io_bazelrc.sh

@@ -65,13 +65,19 @@ build --action_env TF_HEADER_DIR="$PREFIX/lib/python${PY_VER}/site-packages/tens
 build --action_env TF_SHARED_LIBRARY_DIR="$PREFIX/lib/python${PY_VER}/site-packages/tensorflow"
 build --action_env TF_SHARED_LIBRARY_NAME="libtensorflow_framework.so.2"
 build --cxxopt="-std=c++17"
+EOF
 
-# libtirpc support
+# libtirpc support (x86_64 only)
+if [[ "${ARCH}" == "x86_64" ]]; then
+cat >> $BAZEL_RC_DIR/tf_io.bazelrc << EOF
 build --action_env CPATH="$PREFIX/include/tirpc"
 build --action_env LIBRARY_PATH="$PREFIX/lib"
 build --linkopt="-L$PREFIX/lib"
 build --linkopt="-ltirpc"
+EOF
+fi
 
+cat >> $BAZEL_RC_DIR/tf_io.bazelrc << EOF
 build --experimental_repo_remote_exec
 build --enable_platform_specific_config
 build:optimization --compilation_mode=opt

recipe/build-tf-io.sh

@@ -17,11 +17,15 @@
 set -vex
 
 source open-ce-common-utils.sh
+ARCH=$(uname -p)
 
 export TF_PYTHON_VERSION=$PY_VER
-export CFLAGS="-I$PREFIX/include/tirpc $CFLAGS"
-export CXXFLAGS="-I$PREFIX/include/tirpc $CXXFLAGS"
-export LDFLAGS="-L$PREFIX/lib -ltirpc $LDFLAGS"
+
+if [[ "${ARCH}" == "x86_64" ]]; then
+  export CFLAGS="-I$PREFIX/include/tirpc ${CFLAGS}"
+  export CXXFLAGS="-I$PREFIX/include/tirpc ${CXXFLAGS}"
+  export LDFLAGS="-L$PREFIX/lib -ltirpc ${LDFLAGS}"
+fi
 
 # Build Tensorflow from source
 SCRIPT_DIR=$RECIPE_DIR/../buildscripts

recipe/meta.yaml

@@ -31,22 +31,21 @@ outputs:
       build:
         - {{ compiler('c') }}      # [ ppc_arch != "p10"]
         - {{ compiler('cxx') }}    # [ ppc_arch != "p10"]
-        - libtirpc
+        - libtirpc   #[x86_64]
       host:
         - git >=2.2
         - bazel {{ bazel }}
         - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - setuptools {{ setuptools }}
         - python {{ python }}
-        - libtirpc
+        - libtirpc     #[x86_64]
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
         - _tensorflow_select 1.0          #[build_type == 'cpu']
         - _tensorflow_select 2.0          #[build_type == 'cuda']
         - keras {{ keras }}
         - tensorflow-estimator {{ tensorflow }}
         - tensorboard {{ tensorboard }}
-        - libtirpc
       run:
         - python {{ python }}
         - numpy {{ numpy }}
@@ -65,7 +64,7 @@ outputs:
         - {{ compiler('c') }}      # [ ppc_arch != "p10"]
         - {{ compiler('cxx') }}    # [ ppc_arch != "p10"]
         - bazel {{ bazel }}
-        - libtirpc
+        - libtirpc   #[x86_64]
       host:
         - git >=2.2
         - setuptools {{ setuptools }}
@@ -78,7 +77,7 @@ outputs:
         - keras {{ keras }}
         - tensorflow-estimator {{ tensorflow }}
         - tensorboard {{ tensorboard }}
-        - libtirpc
+        - libtirpc     #[x86_64]
       run:
         - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - python {{ python }}