visual-narrator-llm / phase3_cloud_test.py
Ytgetahun's picture
feat: Visual Narrator 3B - Clean repository with professional benchmarks
d6e97b5
import torch
import time
print("๐Ÿš€ PHASE 3 CLOUD VALIDATION TEST")
print("=" * 50)
# Test GPU performance
start_time = time.time()
# Create a large tensor to test GPU memory
x = torch.randn(10000, 10000).cuda()
y = torch.randn(10000, 10000).cuda()
# Perform matrix multiplication (GPU-intensive)
z = torch.matmul(x, y)
end_time = time.time()
print(f"โœ… GPU Performance Test:")
print(f" - Matrix: 10,000 x 10,000")
print(f" - Operation: Matrix Multiplication")
print(f" - Time: {end_time - start_time:.2f} seconds")
print(f" - GPU Memory: {torch.cuda.memory_allocated() / 1024**3:.1f} GB used")
print(f" - A100 Status: READY FOR PHASE 3! ๐ŸŽฏ")
# Clean up
del x, y, z
torch.cuda.empty_cache()