diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-10-27 22:01:51 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-10-27 22:08:32 +0300 |
commit | 429e10dac8cdd4ec1e698d67dde93fea2add7338 (patch) | |
tree | f161a2dd86c511b7866c5464bde7c9022a3e8041 | |
parent | f904e6fe9cf542f5c6f7121f90457fb784f1b388 (diff) |
Enable '*' for CudaDoubleTensors.
-rw-r--r-- | generic/TensorOperator.c | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/generic/TensorOperator.c b/generic/TensorOperator.c index 5844a9d..ddd50a4 100644 --- a/generic/TensorOperator.c +++ b/generic/TensorOperator.c @@ -145,8 +145,7 @@ static int cutorch_TensorOperator_(__unm__)(lua_State *L) static int cutorch_TensorOperator_(__mul__)(lua_State *L) { - // FIXME: implement -#ifdef THC_REAL_IS_FLOAT +#if defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE) THCTensor *tensor1 = luaT_toudata(L, 1, torch_Tensor); THCTensor *tensor2 = luaT_toudata(L, 2, torch_Tensor); THCTensor *r; |