Skip to content

Instantly share code, notes, and snippets.

View ZD-AI-Lab's full-sized avatar

ZD-AI-Lab

  • 至顶科技
  • Beijing
View GitHub Profile
"""
Advanced FP4 Performance Test for NVIDIA B200
This script attempts to access FP4 formats if available
"""
import time
import torch
print(f"PyTorch version: {torch.__version__}")
print(f"CUDA available: {torch.cuda.is_available()}")