Travis Johnson 593e79e733

[Bugfix] torch.set_num_threads() in multiproc_gpu_executor (#6802) ...

[Bugfix] Use torch.set_num_threads() to configure parallelism in multiproc_gpu_executor (#6802)
Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>

2024-07-26 22:15:20 -07:00

..

__init__.py

Add distributed model executor abstraction (#3191)

2024-03-11 11:03:45 -07:00

cpu_executor.py

[Hardware] [Intel] Enable Multiprocessing and tensor parallel in CPU backend and update documentation (#6125)

2024-07-26 13:50:10 -07:00

distributed_gpu_executor.py

[Core] Introduce SPMD worker execution using Ray accelerated DAG (#6032)

2024-07-17 22:27:09 -07:00

executor_base.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

gpu_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

multiproc_gpu_executor.py

[Bugfix] torch.set_num_threads() in multiproc_gpu_executor (#6802)

2024-07-26 22:15:20 -07:00

multiproc_worker_utils.py

[Core][Doc] Default to multiprocessing for single-node distributed case (#5230)

2024-06-11 11:10:41 -07:00

neuron_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

openvino_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

ray_gpu_executor.py

[Core] Fix ray forward_dag error mssg (#6792)

2024-07-25 16:53:25 -07:00

ray_tpu_executor.py

[Hardware][TPU] Implement tensor parallelism with Ray (#5871)

2024-07-26 20:54:27 -07:00

ray_utils.py

[Misc][TPU] Support TPU in initialize_ray_cluster (#6812)

2024-07-26 19:39:49 +00:00

ray_xpu_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

tpu_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00

xpu_executor.py

[Core] Allow specifying custom Executor (#6557)

2024-07-20 01:25:06 +00:00