@@ -9,14 +9,14 @@ set -exu
99
1010CUDA_VERSION=${1:- " 12.6" }
1111
12- echo " === Testing ExecutorTorch CUDA ${CUDA_VERSION} Build ==="
12+ echo " === Testing ExecuTorch CUDA ${CUDA_VERSION} Build ==="
1313
14- # Function to build and test ExecutorTorch with CUDA support
14+ # Function to build and test ExecuTorch with CUDA support
1515test_executorch_cuda_build () {
1616 local cuda_version=$1
1717
18- echo " Building ExecutorTorch with CUDA ${cuda_version} support..."
19- echo " ExecutorTorch will automatically detect CUDA and install appropriate PyTorch wheel"
18+ echo " Building ExecuTorch with CUDA ${cuda_version} support..."
19+ echo " ExecuTorch will automatically detect CUDA and install appropriate PyTorch wheel"
2020
2121 # Check available resources before starting
2222 echo " === System Information ==="
@@ -27,11 +27,11 @@ test_executorch_cuda_build() {
2727 nvcc --version || echo " nvcc not found"
2828 nvidia-smi || echo " nvidia-smi not found"
2929
30- # Set CMAKE_ARGS to enable CUDA build - ExecutorTorch will handle PyTorch installation automatically
30+ # Set CMAKE_ARGS to enable CUDA build - ExecuTorch will handle PyTorch installation automatically
3131 export CMAKE_ARGS=" -DEXECUTORCH_BUILD_CUDA=ON"
3232
33- echo " === Starting ExecutorTorch Installation ==="
34- # Install ExecutorTorch with CUDA support with timeout and error handling
33+ echo " === Starting ExecuTorch Installation ==="
34+ # Install ExecuTorch with CUDA support with timeout and error handling
3535 timeout 5400 ./install_executorch.sh || {
3636 local exit_code=$?
3737 echo " ERROR: install_executorch.sh failed with exit code: $exit_code "
@@ -41,15 +41,15 @@ test_executorch_cuda_build() {
4141 exit $exit_code
4242 }
4343
44- echo " SUCCESS: ExecutorTorch CUDA build completed"
44+ echo " SUCCESS: ExecuTorch CUDA build completed"
4545
4646 # Verify the installation
47- echo " === Verifying ExecutorTorch CUDA Installation ==="
47+ echo " === Verifying ExecuTorch CUDA Installation ==="
4848
49- # Test that ExecutorTorch was built successfully
49+ # Test that ExecuTorch was built successfully
5050 python -c "
5151import executorch
52- print('SUCCESS: ExecutorTorch imported successfully')
52+ print('SUCCESS: ExecuTorch imported successfully')
5353"
5454
5555 # Test CUDA availability and show details
6060 print('INFO: CUDA available:', torch.cuda.is_available())
6161
6262 if torch.cuda.is_available():
63- print('SUCCESS: CUDA is available for ExecutorTorch ')
63+ print('SUCCESS: CUDA is available for ExecuTorch ')
6464 print('INFO: CUDA version:', torch.version.cuda)
6565 print('INFO: GPU device count:', torch.cuda.device_count())
6666 print('INFO: Current GPU device:', torch.cuda.current_device())
7474 print('SUCCESS: CUDA tensor operation completed on device:', z.device)
7575 print('INFO: Result tensor shape:', z.shape)
7676
77- print('SUCCESS: ExecutorTorch CUDA integration verified')
77+ print('SUCCESS: ExecuTorch CUDA integration verified')
7878 else:
79- print('WARNING: CUDA not detected, but ExecutorTorch built successfully')
79+ print('WARNING: CUDA not detected, but ExecuTorch built successfully')
8080 exit(1)
8181except Exception as e:
82- print('ERROR: ExecutorTorch CUDA test failed:', e)
82+ print('ERROR: ExecuTorch CUDA test failed:', e)
8383 exit(1)
8484"
8585
86- echo " SUCCESS: ExecutorTorch CUDA ${cuda_version} build and verification completed successfully"
86+ echo " SUCCESS: ExecuTorch CUDA ${cuda_version} build and verification completed successfully"
8787}
8888
8989# Main execution
0 commit comments