.. |
__init__.py
|
a3e26391e4
chore: add a wrapper for torch.inference_mode decorator
|
hace 5 meses |
cuda.py
|
1a40bf438b
fix: incorrect gpu capability when used mixed gpus
|
hace 5 meses |
interface.py
|
a3e26391e4
chore: add a wrapper for torch.inference_mode decorator
|
hace 5 meses |
rocm.py
|
cda0e93a10
abstract away the platform for device capability
|
hace 5 meses |
tpu.py
|
a3e26391e4
chore: add a wrapper for torch.inference_mode decorator
|
hace 5 meses |