Re-enable Kineto (#4157)

Signed-off-by: Pavel Chekin <[email protected]>

Author: pbchekin

.github/actions/setup-pytorch/action.yml

@@ -126,7 +126,7 @@ runs:
         # CMake Error at third_party/protobuf/cmake/CMakeLists.txt:2 (cmake_minimum_required)
         pip install 'cmake<4.0.0'
         pip install -r requirements.txt
-        USE_KINETO=0 USE_XCCL=1 USE_STATIC_MKL=1 CFLAGS="-Wno-error=maybe-uninitialized" python setup.py bdist_wheel 2>&1 | grep -v \
+        USE_XCCL=1 USE_STATIC_MKL=1 CFLAGS="-Wno-error=maybe-uninitialized" python setup.py bdist_wheel 2>&1 | grep -v \
           "Double arithmetic operation is not supported on this platform with FP64 conversion emulation mode (poison FP64 kernels is enabled)." | grep -v '^$'
 
     - name: Install PyTorch (built from source)

scripts/install-pytorch.sh

@@ -199,7 +199,7 @@ function build_pytorch {
   pip install 'cmake<4.0.0'
   pip install -r requirements.txt
   pip install cmake ninja
-  USE_KINETO=0 USE_XCCL=1 USE_STATIC_MKL=1 python setup.py bdist_wheel
+  USE_XCCL=1 USE_STATIC_MKL=1 python setup.py bdist_wheel
 }
 
 function install_pytorch {

scripts/patch-pytorch.sh

@@ -36,3 +36,8 @@ echo "Applying PyTorch patches in $REPO_ROOT"
 
 # put your patch applies here
 apply_patch https://github.com/pytorch/pytorch/pull/143553.diff
+
+# https://github.com/pytorch/pytorch/pull/152007
+fetch_patch https://patch-diff.githubusercontent.com/raw/pytorch/kineto/pull/1066.diff
+cd "$REPO_ROOT/third_party/kineto"
+git apply "$SCRIPTS_DIR/1066.diff"