Skip to content

Instantly share code, notes, and snippets.

@wonjoolee95
Created January 6, 2023 02:29
Show Gist options
  • Save wonjoolee95/57e12cb6947e56538e0cde23b261021f to your computer and use it in GitHub Desktop.
Save wonjoolee95/57e12cb6947e56538e0cde23b261021f to your computer and use it in GitHub Desktop.
(base) jenkins@26d7adccbc26:/workspace/tmp$ cat test_simple_model_func_1.hlo
[ScheduleSyncTensorsGraph]
TensorsGraphInfo:
test_simple_model (test/dynamo/test_dynamo.py:39)
run (/opt/conda/lib/python3.7/unittest/case.py:628)
__call__ (/opt/conda/lib/python3.7/unittest/case.py:676)
run (/opt/conda/lib/python3.7/unittest/suite.py:122)
__call__ (/opt/conda/lib/python3.7/unittest/suite.py:84)
run (/opt/conda/lib/python3.7/unittest/suite.py:122)
__call__ (/opt/conda/lib/python3.7/unittest/suite.py:84)
run (/opt/conda/lib/python3.7/unittest/runner.py:176)
runTests (/opt/conda/lib/python3.7/unittest/main.py:271)
__init__ (/opt/conda/lib/python3.7/unittest/main.py:101)
<module> (test/dynamo/test_dynamo.py:87)
Hashes: (d88504b1b3dc9ace690ef1218a1c161e)
## BEGIN_GRAPH
HloModule IrToHlo.9, entry_computation_layout={(f32[],f32[])->(f32[])}
ENTRY %IrToHlo.9 (p0.2: f32[], p1.5: f32[]) -> (f32[]) {
%p1.5 = f32[] parameter(1), metadata={op_type="xla__device_data" op_name="xla__device_data" source_file="test_simple_model@test_dynamo.py" source_line=34}
%cosine.6 = f32[] cosine(f32[] %p1.5), metadata={op_type="aten__cos" op_name="aten__cos" source_file="fn_simple@test_dynamo.py" source_line=17}
%p0.2 = f32[] parameter(0), metadata={op_type="xla__device_data" op_name="xla__device_data" source_file="test_simple_model@test_dynamo.py" source_line=35}
%sine.3 = f32[] sine(f32[] %p0.2), metadata={op_type="aten__sin" op_name="aten__sin" source_file="fn_simple@test_dynamo.py" source_line=18}
%constant.1 = f32[] constant(1), metadata={op_type="prim__Constant" op_name="prim__Constant" source_file="fn_simple@test_dynamo.py" source_line=19}
%multiply.4 = f32[] multiply(f32[] %sine.3, f32[] %constant.1), metadata={op_type="aten__mul" op_name="aten__mul" source_file="fn_simple@test_dynamo.py" source_line=19}
%add.7 = f32[] add(f32[] %cosine.6, f32[] %multiply.4), metadata={op_type="aten__add" op_name="aten__add" source_file="fn_simple@test_dynamo.py" source_line=19}
ROOT %tuple.8 = (f32[]) tuple(f32[] %add.7)
}
## END_GRAPH
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment