| 8 | namespace test { |
| 9 | |
| 10 | TEST_F(CAMBRICON, UNIFORM_RNG_F32) { |
| 11 | auto opr = handle_cambricon()->create_operator<UniformRNG>(); |
| 12 | opr->param().dtype = DTypeTrait<dtype::Float32>::enumv; |
| 13 | SyncedTensor<> t(handle_cambricon(), {TensorShape{200000}, dtype::Float32()}); |
| 14 | opr->exec(t.tensornd_dev(), {}); |
| 15 | assert_uniform_correct(t.ptr_mutable_host(), t.layout().total_nr_elems()); |
| 16 | } |
| 17 | |
| 18 | TEST_F(CAMBRICON, GAUSSIAN_RNG_F32) { |
| 19 | auto opr = handle_cambricon()->create_operator<GaussianRNG>(); |
nothing calls this directly
no test coverage detected