Commit d88ca899 authored by Brian Pickrell's avatar Brian Pickrell
Browse files

data type in test

parent 9353b24a
...@@ -94,7 +94,7 @@ struct rand_uniform ...@@ -94,7 +94,7 @@ struct rand_uniform
argument compute(const shape& output, std::vector<argument> args) const argument compute(const shape& output, std::vector<argument> args) const
{ {
// Output goes into the passed buffer, not the shape output // Output goes into the passed buffer, not the shape output
(void) output; (void)output;
argument result{args[1].get_shape()}; argument result{args[1].get_shape()};
uint32_t local_seed = args[0].at<uint32_t>(0); uint32_t local_seed = args[0].at<uint32_t>(0);
......
...@@ -2219,7 +2219,7 @@ TEST_CASE(prefix_scan_sum_dyn_2d) ...@@ -2219,7 +2219,7 @@ TEST_CASE(prefix_scan_sum_dyn_2d)
TEST_CASE(rand_uniform) TEST_CASE(rand_uniform)
{ {
std::vector<migraphx::shape::dynamic_dimension> dd{{5, 8}, {3, 7}}; std::vector<migraphx::shape::dynamic_dimension> dd{{5, 8}, {3, 7}};
migraphx::shape s0{migraphx::shape::float_type, {1}}; migraphx::shape s0{migraphx::shape::uint32_type, {1}};
migraphx::shape s1{migraphx::shape::float_type, dd}; migraphx::shape s1{migraphx::shape::float_type, dd};
expect_shape(s1, migraphx::make_op("rand_uniform"), s0, s1); expect_shape(s1, migraphx::make_op("rand_uniform"), s0, s1);
} }
......
...@@ -6537,7 +6537,7 @@ TEST_CASE(rand_uniform_and_seed_test) ...@@ -6537,7 +6537,7 @@ TEST_CASE(rand_uniform_and_seed_test)
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
float control_seed = 0.0f; float control_seed = 0.0f;
size_t sample_size(20000); size_t sample_size(20000);
// Shape of the random data // Shape of the random data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment