Home
last modified time | relevance | path

Searched full:multigpu (Results 1 – 15 of 15) sorted by relevance

/external/pytorch/.github/workflows/
D_rocm-test.yml173 if [[ $TEST_CONFIG == 'multigpu' ]]; then
174 TEST_COMMAND=.ci/pytorch/multigpu-test.sh
D_linux-test.yml218 if [[ $TEST_CONFIG == 'multigpu' ]]; then
219 TEST_COMMAND=.ci/pytorch/multigpu-test.sh
Dperiodic.yml127 … { config: "multigpu", shard: 1, num_shards: 1, runner: "am2.linux.g5.12xlarge.nvidia.gpu" },
/external/pytorch/test/
Dtest_numba_integration.py217 @unittest.skipIf(not TEST_MULTIGPU, "No multigpu")
373 @unittest.skipIf(not TEST_MULTIGPU, "No multigpu")
Dtest_spectral_ops.py864 # Multigpu tests
/external/pytorch/.github/actions/linux-test/
Daction.yml188 if [[ $TEST_CONFIG == 'multigpu' ]]; then
189 TEST_COMMAND=.ci/pytorch/multigpu-test.sh
/external/pytorch/docs/source/notes/
Damp_examples.rst287 .. _amp-multigpu:
/external/pytorch/torch/testing/_internal/
Dcommon_distributed.py459 # HELPER FOR MULTIGPU TESTS
461 """Multigpu tests are designed to simulate the multi nodes with multi
/external/pytorch/torch/amp/
Dautocast_mode.py202 (see :ref:`Working with Multiple GPUs<amp-multigpu>`).
/external/pytorch/test/distributed/
Dtest_store.py58 """Multigpu tests are designed to simulate the multi nodes with multi
Dtest_c10d_common.py64 """Multigpu tests are designed to simulate the multi nodes with multi
/external/angle/src/third_party/libXNVCtrl/
DNVCtrl.h2790 * draw information about the current MULTIGPU mode.
/external/aws-sdk-java-v2/services/neptunedata/src/main/resources/codegen-resources/
Dservice-2.json4114 …ce used for model training. All Neptune ML models support CPU, GPU, and multiGPU training. The def…
/external/pytorch/.github/scripts/
Drockset_mocks.json.gz
Dgql_mocks.json.gz1{ 2 "query_sha=e3d02ae0630578ecbdc5a012ecae959d0387a6d0f642a72bc4fc46dc9ee99622 name=pytorch number=105312 owner=pytorch": { 3 "data" ...