nicholas-leonard / equanimity

Experimental research for distributed conditional computation
4 stars 0 forks source link

cutorch:index #33

Closed nicholas-leonard closed 10 years ago

nicholas-leonard commented 10 years ago

torch: https://github.com/torch/torch7/blob/master/lib/TH/generic/THTensorMath.c#L69 https://github.com/torch/torch7/blob/master/generic/Tensor.c#L387 cutorch: https://github.com/torch/cutorch/blob/master/lib/THC/THCTensorMath.cu https://github.com/torch/cutorch/blob/master/lib/THC/THCTensorCopy.cu

nicholas-leonard commented 10 years ago

@hycis found that this already works :P