Run script in workflow

Author: Rohan-Bierneni

.github/workflows/UnitTests.yml

@@ -101,6 +101,34 @@ jobs:
       - name: Check devices
         run: |
           python -c "import jax; print(jax.devices())"
+      
+      - name: Run Conflict Verification Script
+        run: |
+          # This command creates the file inside the runner
+          cat <<'EOF' > verify_conflict.py
+          print("--- PyTorch vs. JAX Conflict Test ---")
+
+          print("\nStep 1: Attempting to import torch...")
+          try:
+              import torch
+              print(f"Successfully imported torch version: {torch.__version__}")
+              print(f"Is PyTorch using CUDA? -> {torch.cuda.is_available()}")
+          except Exception as e:
+              print(f"Failed to import torch: {e}")
+
+          print("\nStep 2: Now, attempting to initialize JAX...")
+          try:
+              import jax
+              devices = jax.devices()
+              print("\n--- RESULT: SUCCESS ---")
+              print(f"JAX initialized correctly and found devices: {devices}")
+          except Exception as e:
+              print("\n--- RESULT: FAILURE ---")
+              print("JAX failed to initialize after PyTorch was imported.")
+              print(f"JAX Error: {e}")
+          EOF
+
+          # Now that the file exists, this command will work
           python verify_conflict.py
 
       # - name: Run MaxDiffusion Training