| commit | 233017f01f0f7ad05df886276864d8587b0aa43f | [log] [tgz] |
|---|---|---|
| author | Adam Paszke <adam.paszke@gmail.com> | Sat Nov 19 15:25:59 2016 +0100 |
| committer | Adam Paszke <adam.paszke@gmail.com> | Mon Nov 21 19:39:55 2016 +0100 |
| tree | 5cc5ff69a75485401766e5bc4b13f45bbc9b3f65 | |
| parent | 0613ac90cd1201b7a0c73f86365368aba7f0b096 [diff] |
Add torch.multinomial for CUDA
diff --git a/torch/csrc/generic/methods/TensorRandom.cwrap b/torch/csrc/generic/methods/TensorRandom.cwrap index d45acf0..4bd69ae 100644 --- a/torch/csrc/generic/methods/TensorRandom.cwrap +++ b/torch/csrc/generic/methods/TensorRandom.cwrap
@@ -171,6 +171,7 @@ [[ name: multinomial defined_if: CUDA_FLOAT || CUDA_DOUBLE || CUDA_HALF + with_stateless: True return: argument 0 arguments: - arg: THTensor* result