diff options
| author | soumith <soumith@fb.com> | 2016-08-11 01:26:11 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2016-08-11 01:26:11 +0300 |
| commit | e9a349b6f710804511c0fc4fc1a20dfb37b943e2 (patch) | |
| tree | be70e8124932c5af4727c8a05b19cf01fc036466 /TensorMath.lua | |
| parent | 7b4dc42d02ee1d7ad44b25babe86b2d2ca0b336e (diff) | |
making changes to sort and TopK for the changed index* API
Diffstat (limited to 'TensorMath.lua')
| -rw-r--r-- | TensorMath.lua | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/TensorMath.lua b/TensorMath.lua index 7c08542..2f16545 100644 --- a/TensorMath.lua +++ b/TensorMath.lua @@ -1063,20 +1063,20 @@ wrap("scatter", wrap("sort", cname("sort"), {{name=Tensor, default=true, returned=true}, - {name=Tensor, default=true, returned=true, noreadadd=true}, - {name=Tensor}, - {name="index", default=lastdim(3)}, - {name="boolean", default=0}}) + {name="CudaLongTensor", default=true, returned=true, noreadadd=true}, + {name=Tensor}, + {name="index", default=lastdim(3)}, + {name="boolean", default=0}}) wrap("topk", cname("topk"), {{name=Tensor, default=true, returned=true}, - {name=Tensor, default=true, returned=true, noreadadd=true}, - {name=Tensor}, - {name="long", default=1}, - {name="index", default=lastdim(3)}, - {name="boolean", default=0}, - {name="boolean", default=0}}) + {name="CudaLongTensor", default=true, returned=true, noreadadd=true}, + {name=Tensor}, + {name="long", default=1}, + {name="index", default=lastdim(3)}, + {name="boolean", default=0}, + {name="boolean", default=0}}) do local Tensor = Tensor |