MCPcopy Index your code
hub / github.com/OpenDriveLab/ReSim / test_model_parallel.py

File test_model_parallel.py

SwissArmyTransformer/tests/test_model_parallel.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1from sat.mpu import mp_split_model
2from sat.model import BertModel
3import torch
4import os

Callers

nothing calls this directly

Calls 2

test_bert_inferenceFunction · 0.85
printFunction · 0.50

Tested by

no test coverage detected