Austin Veselka eefeb16464

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524) ...

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>

2024-04-27 00:03:48 -07:00

..

bgmv_bf16_bf16_bf16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_bf16_fp32_bf16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_config.h

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_fp16_fp16_fp16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_fp16_fp32_fp16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_fp32_bf16_bf16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_fp32_fp16_fp16.cu

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

bgmv_impl.cuh

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

generator.py

[Kernel] Full Tensor Parallelism for LoRA Layers (#3524)

2024-04-27 00:03:48 -07:00

vec_dtypes.cuh

[Experimental] Add multi-LoRA support (#1804)

2024-01-23 15:26:37 -08:00