Sign in
android
/
platform
/
external
/
pytorch
/
cf55670bdd
/
.
/
torch
/
lib
/
THD
/
base
/
Cuda.h
blob: 0442fdc3ac6ad4439f3abe788c99a0c23fcca5c6 [
file
] [
log
] [
blame
]
#pragma
once
#ifdef
USE_CUDA
#include
<THD/THD.h>
#include
<THC/THC.h>
THD_API
void
THDSetCudaStatePtr
(
THCState
**
state
);
THD_API
void
THDRegisterCudaStream
(
cudaStream_t
stream
);
#endif