mirror of
https://github.com/marian-nmt/marian.git
synced 2024-08-15 01:00:23 +03:00
towards cudnn
This commit is contained in:
parent
a43d0b27c0
commit
4291c918ae
@ -1,6 +1,12 @@
|
||||
#include <iostream>
|
||||
#include <ctime>
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
#include <device_launch_parameters.h>
|
||||
|
||||
#include <cublas_v2.h>
|
||||
#include <cudnn.h>
|
||||
|
||||
#include "marian.h"
|
||||
|
||||
using namespace marian;
|
||||
|
Loading…
Reference in New Issue
Block a user