From 762929a2b7b8458b8228f3738aa97a2b1d740659 Mon Sep 17 00:00:00 2001
From: Jae Yoo <jaeyoo@google.com>
Date: Tue, 14 Mar 2023 17:57:59 -0700
Subject: [PATCH] Upgrade to TF 2.11.0

---
 .bazelrc | 12 ++++++++++++
 setup.py |  4 ++--
 2 files changed, 14 insertions(+), 2 deletions(-)
 create mode 100644 .bazelrc

diff --git a/.bazelrc b/.bazelrc
new file mode 100644
index 0000000..f974751
--- /dev/null
+++ b/.bazelrc
@@ -0,0 +1,12 @@
+build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true
+build:manylinux2010cuda100 --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain
+build:manylinux2010cuda101 --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain
+build --spawn_strategy=standalone
+build --strategy=Genrule=standalone
+build -c opt
+build --action_env TF_HEADER_DIR="/usr/local/google/home/jaeyoo/workspace/cpu_venv/lib/python3.8/site-packages/tensorflow/include"
+build --action_env TF_SHARED_LIBRARY_DIR="/usr/local/google/home/jaeyoo/workspace/cpu_venv/lib/python3.8/site-packages/tensorflow"
+build --action_env TF_SHARED_LIBRARY_NAME="libtensorflow_framework.so.2"
+build --action_env TF_NEED_CUDA="0"
+build --config=manylinux2010cuda101
+test --config=manylinux2010cuda101
diff --git a/setup.py b/setup.py
index 69245d6..ed85d22 100644
--- a/setup.py
+++ b/setup.py
@@ -23,9 +23,9 @@
 from setuptools.dist import Distribution
 
 
-__version__ = '0.0.1'
+__version__ = '0.0.2'
 REQUIRED_PACKAGES = [
-    'tensorflow >= 2.1.0',
+    'tensorflow == 2.11.0',
 ]
 project_name = 'tensorflow-custom-ops'