use integration test as the badge shown on the homepage

ghstack-source-id: 775591945ff5427cb7e5e9fc7592952b4c746341
Pull Request resolved: #373

Author: tianyu-l

.github/workflows/integration_test_periodic.yaml .github/workflows/integration_test_4gpu.yaml

@@ -1,6 +1,9 @@
-name: GPU Integration Test
+name: 4 GPU Integration Test
 
 on:
+  push:
+    branches: [ main ]
+  pull_request:
   schedule:
     # Runs hourly
     - cron: '0 * * * *'

.github/workflows/unit_test_4gpu.yaml

@@ -1,4 +1,4 @@
-[![GPU Integration Test](https://github.com/pytorch/torchtitan/actions/workflows/unit_test_4gpu.yaml/badge.svg?branch=main)](https://github.com/pytorch/torchtitan/actions/workflows/unit_test_4gpu.yaml)
+[![4 GPU Integration Test](https://github.com/pytorch/torchtitan/actions/workflows/integration_test_4gpu.yaml/badge.svg?branch=main)](https://github.com/pytorch/torchtitan/actions/workflows/integration_test_4gpu.yaml)
 
 # torchtitan
 

README.md

@@ -3,7 +3,6 @@
 [job]
 dump_folder = "./outputs"
 description = "Llama 3 debug training"
-# TODO: turn this back on once ci have tokenizer
 use_for_integration_test = true
 
 [profiling]
@@ -50,7 +49,7 @@ interval_type = "steps"
 interval = 5
 model_weights_only = false
 export_dtype = "float32"
-async_mode = "disabled" # ["disabled", "async", "async_with_pinned_mem"]
+async_mode = "disabled"  # ["disabled", "async", "async_with_pinned_mem"]
 
 [activation_checkpoint]
 mode = 'selective'  # ['none', 'selective', 'full']