Fix layernorm verify test

src/targets/gpu/jit/bert_transpose.cpp

@@ -40,8 +40,9 @@ struct transposectx_compiler : compiler<transposectx_compiler>
     operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const
     {
         hip_compile_options options;
+        auto h = inputs.front().lens().back();
         options.set_launch_params(
-            v, compute_global_for(ctx, inputs.back().elements()), inputs.front().lens().back());
+            v, compute_global_for(ctx, inputs.back().elements(), h), h);
         options.output      = inputs.back();
         options.inputs      = inputs;
         options.kernel_name = "transposectx_kernel";
@@ -79,8 +80,9 @@ struct transposeqkv_compiler : compiler<transposeqkv_compiler>
     operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const
     {
         hip_compile_options options;
+        auto h = inputs.front().lens().back();
         options.set_launch_params(
-            v, compute_global_for(ctx, inputs.back().elements()), inputs.front().lens().back());
+            v, compute_global_for(ctx, inputs.back().elements(), h), h);
         options.output      = inputs.back();
         options.inputs      = inputs;
         options.kernel_name = "transposeqkv_kernel";

test/verify/0layernorm_test.cpp

@@ -11,7 +11,7 @@ struct test_layernorm : verify_program<test_layernorm>
         auto* mm = p.get_main_module();
         auto x =
             mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 384, 768}});
-        mm->add_instruction(migraphx::make_op("layernorm", {{"axis", -1}}), x);
+        mm->add_instruction(migraphx::make_op("layernorm", {{"axis", -1}, {"epsilon", 1e-12}}), x);
 
         return p;
     }