Thomas ParnellandChih-Chieh-Yang eaec4b9153 [Bugfix] Add custom Triton cache manager to resolve MoE MP issue (#6140) ...

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
Co-authored-by: Chih-Chieh-Yang <chih.chieh.yang@ibm.com>

2024-07-15 10:12:47 -07:00

..

__init__.py

Add distributed model executor abstraction (#3191)

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

cpu_executor.py

[CORE] Adding support for insertion of soft-tuned prompts (#4645)

2024-07-09 13:26:36 -07:00

distributed_gpu_executor.py

[Core] Pipeline Parallel Support (#4412)

2024-07-02 10:58:08 -07:00

executor_base.py

[CORE] Adding support for insertion of soft-tuned prompts (#4645)

2024-07-09 13:26:36 -07:00

gpu_executor.py

[CORE] Adding support for insertion of soft-tuned prompts (#4645)

2024-07-09 13:26:36 -07:00

multiproc_gpu_executor.py

[Bugfix] Add custom Triton cache manager to resolve MoE MP issue (#6140)

2024-07-15 10:12:47 -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

[Bugfix][Neuron] Fix soft prompt method error in NeuronExecutor (#6313)

2024-07-10 16:39:02 -07:00

openvino_executor.py

[Bugfix] Fix OpenVINOExecutor abstractmethod error (#6296)

2024-07-10 10:03:32 -07:00

ray_gpu_executor.py

[ci] try to add multi-node tests (#6280)

2024-07-12 21:51:48 -07:00

ray_utils.py

[BugFix] Avoid unnecessary Ray import warnings (#6079)

2024-07-03 14:09:40 +08:00

ray_xpu_executor.py

[CORE] Adding support for insertion of soft-tuned prompts (#4645)

2024-07-09 13:26:36 -07:00

tpu_executor.py

[Bugfix][TPU] Add prompt adapter methods to TPUExecutor (#6279)

2024-07-09 19:30:56 -07:00

xpu_executor.py

[CORE] Adding support for insertion of soft-tuned prompts (#4645)

2024-07-09 13:26:36 -07:00