Skip to content

Commit 46e7042

Browse files
rguthrie3soumith
authored andcommitted
SIMD helper header, modified add in THTensorMath to check dispatch
1 parent d0c1827 commit 46e7042

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

generic/THTensorMath.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@ accreal THTensor_(prodall)(THTensor *tensor)
460460
return prod;
461461
}
462462

463-
void THTensor_(add)(THTensor *r_, THTensor *t, real value)
463+
void THTensor_(add_Default)(THTensor *r_, THTensor *t, real value)
464464
{
465465
THTensor_(resizeAs)(r_, t);
466466
if (THTensor_(isContiguous)(r_) && THTensor_(isContiguous)(t) && THTensor_(nElement)(r_) == THTensor_(nElement)(t)) {

generic/THTensorMath.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ TH_API accreal THTensor_(prodall)(THTensor *t);
3030
TH_API void THTensor_(neg)(THTensor *self, THTensor *src);
3131
TH_API void THTensor_(cinv)(THTensor *self, THTensor *src);
3232

33-
TH_API void THTensor_(add)(THTensor *r_, THTensor *t, real value);
33+
TH_API void THTensor_(add_Default)(THTensor *r_, THTensor *t, real value);
3434
TH_API void THTensor_(sub)(THTensor *self, THTensor *src, real value);
3535
TH_API void THTensor_(mul)(THTensor *r_, THTensor *t, real value);
3636
TH_API void THTensor_(div)(THTensor *r_, THTensor *t, real value);

0 commit comments

Comments
 (0)