Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save shunting314/853e5926e7fa85ae67bd0ee02bf7df2f to your computer and use it in GitHub Desktop.
Save shunting314/853e5926e7fa85ae67bd0ee02bf7df2f to your computer and use it in GitHub Desktop.
IR {
%0 = s64[] xla::device_data(), device=GPU:0
%1 = s64[] prim::Constant(), value=214013
%2 = s64[] aten::mul(%1, %0)
%3 = s64[] prim::Constant(), value=2531011
%4 = s64[] aten::add(%3, %2)
%5 = s64[] prim::Constant(), value=214013
%6 = s64[] aten::mul(%5, %4)
%7 = s64[] prim::Constant(), value=2531011
%8 = s64[] aten::add(%7, %6)
%9 = s64[] prim::Constant(), value=214013
%10 = s64[] aten::mul(%9, %8)
%11 = s64[] prim::Constant(), value=2531011
%12 = s64[] aten::add(%11, %10)
%13 = s64[] prim::Constant(), value=214013
%14 = s64[] aten::mul(%13, %12)
%15 = s64[] prim::Constant(), value=2531011
%16 = s64[] aten::add(%15, %14)
%17 = f32[] prim::Constant(), value=1
%18 = f32[2,2]{1,0} aten::expand(%17), size=(2, 2)
%19 = f32[] prim::Constant(), value=0
%20 = f32[2,2]{1,0} aten::expand(%19), size=(2, 2)
%21 = f32[2,2]{1,0} aten::normal(%20, %18, %16)
%22 = f32[] prim::Constant(), value=1
%23 = f32[2,2]{1,0} aten::expand(%22), size=(2, 2)
%24 = f32[] prim::Constant(), value=0
%25 = f32[2,2]{1,0} aten::expand(%24), size=(2, 2)
%26 = f32[2,2]{1,0} aten::normal(%25, %23, %12)
%27 = f32[2,2]{1,0} aten::mm(%26, %21), ROOT=0
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment