Add missing RocBlas include (#4557) · 869629c210 - DeepSpeed - Git

Add missing RocBlas include (#4557)

Просмотр исходного кода

This commit is contained in:

Logan Adams

2023-10-23 14:11:04 -07:00

коммит произвёл

GitHub

Родитель a02de228d0

Коммит 869629c210

Не найден ключ, соответствующий данной подписи

Идентификатор ключа GPG: 4AEE18F83AFDEB23

1 изменённых файлов: 3 добавлений и 0 удалений

									
										3

csrc/includes/cublas_wrappers.h

										Просмотреть файл
									
				@ -13,6 +13,9 @@

				#ifndef __HIP_PLATFORM_AMD__

				#include <mma.h>

				#endif

				#ifdef __HIP_PLATFORM_AMD__

				#include <rocblas/rocblas.h>

				#endif

				#include <stdio.h>

				int cublas_gemm_ex(cublasHandle_t handle,