Joe Runde 5002175e80 [Kernel] Add punica dimensions for Granite 13b (#5559) ...

Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>

2024-06-18 03:54:11 +00: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] Add punica dimensions for Granite 13b (#5559)

2024-06-18 03:54:11 +00: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

[ROCm] Add support for Punica kernels on AMD GPUs (#3140)

2024-05-09 09:19:50 -07:00

generator.py

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

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

vec_dtypes.cuh

[ROCm] Add support for Punica kernels on AMD GPUs (#3140)

2024-05-09 09:19:50 -07:00