@@ -9,14 +9,14 @@ set -exu
9
9
10
10
CUDA_VERSION=${1:- " 12.6" }
11
11
12
- echo " === Testing ExecutorTorch CUDA ${CUDA_VERSION} Build ==="
12
+ echo " === Testing ExecuTorch CUDA ${CUDA_VERSION} Build ==="
13
13
14
- # Function to build and test ExecutorTorch with CUDA support
14
+ # Function to build and test ExecuTorch with CUDA support
15
15
test_executorch_cuda_build () {
16
16
local cuda_version=$1
17
17
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"
20
20
21
21
# Check available resources before starting
22
22
echo " === System Information ==="
@@ -27,11 +27,11 @@ test_executorch_cuda_build() {
27
27
nvcc --version || echo " nvcc not found"
28
28
nvidia-smi || echo " nvidia-smi not found"
29
29
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
31
31
export CMAKE_ARGS=" -DEXECUTORCH_BUILD_CUDA=ON"
32
32
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
35
35
timeout 5400 ./install_executorch.sh || {
36
36
local exit_code=$?
37
37
echo " ERROR: install_executorch.sh failed with exit code: $exit_code "
@@ -41,15 +41,15 @@ test_executorch_cuda_build() {
41
41
exit $exit_code
42
42
}
43
43
44
- echo " SUCCESS: ExecutorTorch CUDA build completed"
44
+ echo " SUCCESS: ExecuTorch CUDA build completed"
45
45
46
46
# Verify the installation
47
- echo " === Verifying ExecutorTorch CUDA Installation ==="
47
+ echo " === Verifying ExecuTorch CUDA Installation ==="
48
48
49
- # Test that ExecutorTorch was built successfully
49
+ # Test that ExecuTorch was built successfully
50
50
python -c "
51
51
import executorch
52
- print('SUCCESS: ExecutorTorch imported successfully')
52
+ print('SUCCESS: ExecuTorch imported successfully')
53
53
"
54
54
55
55
# Test CUDA availability and show details
60
60
print('INFO: CUDA available:', torch.cuda.is_available())
61
61
62
62
if torch.cuda.is_available():
63
- print('SUCCESS: CUDA is available for ExecutorTorch ')
63
+ print('SUCCESS: CUDA is available for ExecuTorch ')
64
64
print('INFO: CUDA version:', torch.version.cuda)
65
65
print('INFO: GPU device count:', torch.cuda.device_count())
66
66
print('INFO: Current GPU device:', torch.cuda.current_device())
74
74
print('SUCCESS: CUDA tensor operation completed on device:', z.device)
75
75
print('INFO: Result tensor shape:', z.shape)
76
76
77
- print('SUCCESS: ExecutorTorch CUDA integration verified')
77
+ print('SUCCESS: ExecuTorch CUDA integration verified')
78
78
else:
79
- print('WARNING: CUDA not detected, but ExecutorTorch built successfully')
79
+ print('WARNING: CUDA not detected, but ExecuTorch built successfully')
80
80
exit(1)
81
81
except Exception as e:
82
- print('ERROR: ExecutorTorch CUDA test failed:', e)
82
+ print('ERROR: ExecuTorch CUDA test failed:', e)
83
83
exit(1)
84
84
"
85
85
86
- echo " SUCCESS: ExecutorTorch CUDA ${cuda_version} build and verification completed successfully"
86
+ echo " SUCCESS: ExecuTorch CUDA ${cuda_version} build and verification completed successfully"
87
87
}
88
88
89
89
# Main execution
0 commit comments