Update on "Higher order op for preserving leaf functions through trac…

…e, particularly for getting user defined hooks to compiled autograd"

cc penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx chenyang78 aakhundov kadeng

[ghstack-poisoned]

.ci/pytorch/win-test-helpers/test_custom_script_ops.bat

@@ -26,11 +26,6 @@ popd
 python test_custom_ops.py -v
 if ERRORLEVEL 1 exit /b 1
 
-:: TODO: fix and re-enable this test
-:: See https://github.com/pytorch/pytorch/issues/25155
-:: python test_custom_classes.py -v
-:: if ERRORLEVEL 1 exit /b 1
-
 python model.py --export-script-module="build/model.pt"
 if ERRORLEVEL 1 exit /b 1
 

.ci/pytorch/win-test-helpers/test_libtorch.bat

@@ -1,7 +1,3 @@
-:: Skip LibTorch tests when building a GPU binary and testing on a CPU machine
-:: because LibTorch tests are not well designed for this use case.
-::if "%USE_CUDA%" == "0" IF NOT "%CUDA_VERSION%" == "cpu" exit /b 0
-
 call %SCRIPT_HELPERS_DIR%\setup_pytorch_env.bat
 if errorlevel 1 exit /b 1
 
@@ -34,10 +30,6 @@ set CPP_TESTS_DIR=%TMP_DIR_WIN%\build\torch\test
 :: Skip verify_api_visibility as it a compile level test
 if "%~1" == "verify_api_visibility" goto :eof
 
-if "%~1" == "module_test" goto :eof
-:: See https://github.com/pytorch/pytorch/issues/25312
-if "%~1" == "converter_nomigraph_test" goto :eof
-
 echo Running "%~2"
 if "%~1" == "c10_intrusive_ptr_benchmark" (
   :: NB: This is not a gtest executable file, thus couldn't be handled by pytest-cpp

.github/actionlint.yaml

@@ -7,7 +7,7 @@ self-hosted-runner:
     - linux.4xlarge
     - linux.12xlarge
     - linux.24xlarge
-    - linux.t4g.2xlarge
+    - linux.arm64.2xlarge
     - linux.4xlarge.nvidia.gpu
     - linux.8xlarge.nvidia.gpu
     - linux.16xlarge.nvidia.gpu

.github/ci_commit_pins/fbgemm.txt

@@ -1 +1 @@
-1b2746f642cc2c99fe9d1a0c34359c0de45341c2
+0346155d7f15fbe8be72687e665078edbe1ca5aa

.github/ci_commit_pins/vision.txt

@@ -1 +1 @@
-e13b8f5c3616bdc58fa847a848d63acdd416a692
+1a9ff0d728787df66a372283328158879805dcad