Rate this Page

Computation times#

00:06.395 total execution time for 8 files from unstable:

Example

Time

Mem (MB)

Getting Started with Nested Tensors (nestedtensor.py)

00:06.380

0.0

(prototype) GPU Quantization with TorchAO (gpu_quantization_torchao_tutorial.py)

00:00.003

0.0

MaskedTensor Overview (maskedtensor_overview.py)

00:00.003

0.0

Efficiently writing “sparse” semantics for Adagrad with MaskedTensor (maskedtensor_adagrad.py)

00:00.002

0.0

torch.vmap (vmap_recipe.py)

00:00.002

0.0

MaskedTensor Advanced Semantics (maskedtensor_advanced_semantics.py)

00:00.002

0.0

MaskedTensor Sparsity (maskedtensor_sparsity.py)

00:00.002

0.0

(prototype) Accelerating torch.save and torch.load with GPUDirect Storage (gpu_direct_storage.py)

00:00.000

0.0