Python TPUAccelerator.auto_device_count Examples

Programming Language: Python

Namespace/Package Name: pytorch_lightning.accelerators

Class/Type: TPUAccelerator

Method/Function: auto_device_count

Examples at hotexamples.com: 2

Python TPUAccelerator.auto_device_count - 2 examples found. These are the top rated real world Python examples of pytorch_lightning.accelerators.TPUAccelerator.auto_device_count extracted from open source projects. You can rate examples to help us improve the quality of examples.

Frequently Used Methods

Show Hide

auto_device_count(2)

TPUAccelerator(1)

broadcast(1)

Example #1

Show file

def test_auto_device_count(_):
    assert CPUAccelerator.auto_device_count() == 1
    assert CUDAAccelerator.auto_device_count() == 2
    assert TPUAccelerator.auto_device_count() == 8
    assert IPUAccelerator.auto_device_count() == 4

Example #2

Show file

File: test_common.py Project: wandb/pytorch-lightning

def test_auto_device_count(device_count_mock):
    assert CPUAccelerator.auto_device_count() == 1
    assert GPUAccelerator.auto_device_count() == 2
    assert TPUAccelerator.auto_device_count() == 8
    assert IPUAccelerator.auto_device_count() == 4