| 52 | } // anonymous namespace |
| 53 | |
| 54 | TEST(TestOprElemwiseMultiType, Fma3Int16x32x32x32) { |
| 55 | make_elemwise_multi_type_checker3( |
| 56 | opr::ElemwiseMultiType::Mode::FUSE_MUL_ADD3_INT16x32x32x32, |
| 57 | {dtype::Int16{}, dtype::Int32{}, dtype::Int32{}}) |
| 58 | ->disable_grad_check() |
| 59 | .run({TensorShape{3, 4, 5}, {1, 4, 1}, {1, 4, 1}}) |
| 60 | .run({TensorShape{1, 4, 5}, {1, 4, 1}, {1, 4, 1}}) |
| 61 | .run({TensorShape{3, 4, 5}, {3, 4, 1}, {3, 4, 1}}); |
| 62 | } |
| 63 | |
| 64 | TEST(TestOprElemwiseMultiType, Fma3IXxf32xf32xi8) { |
| 65 | std::array<DType, 3> src_types{dtype::Int8{}, dtype::Int16{}, dtype::Int32{}}; |
nothing calls this directly
no test coverage detected