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