MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / test_create_dyn_shape

Function test_create_dyn_shape

test/py/test_shape.py:81–89  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

79
80
81def test_create_dyn_shape():
82 a = migraphx.shape.dynamic_dimension(1, 4, {2, 4})
83 b = migraphx.shape.dynamic_dimension(4, 4)
84 dds = [a, b]
85 dyn_shape = migraphx.shape(type='float', dyn_dims=dds)
86 assert dyn_shape.dynamic()
87 assert dyn_shape.dyn_dims()[0].min == dds[0].min
88 assert dyn_shape.dyn_dims()[0].max == dds[0].max
89 assert dyn_shape.dyn_dims()[0].optimals == dds[0].optimals
90
91
92def test_type_enum():

Callers 1

test_shape.pyFile · 0.85

Calls 2

dynamicMethod · 0.80
shapeMethod · 0.45

Tested by

no test coverage detected