guards for half

1 files changed, 8 insertions, 0 deletions

diff --git a/init.c b/init.c
index b5c0f05..184027a 100644
--- a/init.c
+++ b/init.c
@@ -901,13 +901,21 @@ static int cutorch_shutdown(lua_State *L)
 
 static int cutorch_hasHalfInstructions(lua_State *L) {
   THCState *state = cutorch_getstate(L);
+#ifdef CUDA_HALF_TENSOR
   lua_pushboolean(L, THC_nativeHalfInstructions(state));
+#else
+  lua_pushboolean(L, false);
+#endif
   return 1;
 }
 
 static int cutorch_hasFastHalfInstructions(lua_State *L) {
   THCState *state = cutorch_getstate(L);
+#ifdef CUDA_HALF_TENSOR  
   lua_pushboolean(L, THC_fastHalfInstructions(state));
+#else
+  lua_pushboolean(L, false);
+#endif  
   return 1;
 }