pytorch

Форк
0
/
cuda_dlink_extension_add.cu 
6 строк · 139.0 Байт
1
#include <cuda.h>
2
#include <cuda_runtime.h>
3

4
__device__ void add(const float* a, const float* b, float* output) {
5
    *output = *a + *b;
6
}
7

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.