| 856 | } |
| 857 | |
| 858 | void testAllBroadcast(dim4 dims) { |
| 859 | af::array A = constant(1, dims); |
| 860 | for (int k = 0; k < dims.ndims(); ++k) { |
| 861 | dim4 rdims = dims; |
| 862 | rdims[k] = 1; |
| 863 | af::array B = constant(-1, rdims); |
| 864 | af::array C = A + B; |
| 865 | ASSERT_ARRAYS_EQ(C, constant(0, dims)); |
| 866 | |
| 867 | C = B + A; |
| 868 | ASSERT_ARRAYS_EQ(C, constant(0, dims)); |
| 869 | } |
| 870 | } |
| 871 | |
| 872 | TEST(Broadcast, MatrixMatrix2d) { testAllBroadcast(dim4(10, 15)); } |
| 873 |