| 29 | |
| 30 | |
| 31 | def get_test_program() -> Program: |
| 32 | return Program( |
| 33 | version=0, |
| 34 | execution_plan=[ |
| 35 | ExecutionPlan( |
| 36 | name="forward", |
| 37 | values=[ |
| 38 | EValue(Int(1)), |
| 39 | EValue(Int(0)), |
| 40 | EValue(Null()), |
| 41 | EValue(String("pass")), |
| 42 | EValue( |
| 43 | val=Tensor( |
| 44 | scalar_type=ScalarType.FLOAT, |
| 45 | storage_offset=0, |
| 46 | sizes=[2, 2], |
| 47 | dim_order=[0, 1], |
| 48 | requires_grad=False, |
| 49 | layout=0, |
| 50 | data_buffer_idx=0, |
| 51 | allocation_info=AllocationDetails( |
| 52 | memory_id=1, |
| 53 | memory_offset_high=0, |
| 54 | memory_offset_low=16, |
| 55 | ), |
| 56 | shape_dynamism=TensorShapeDynamism.STATIC, |
| 57 | ) |
| 58 | ), |
| 59 | ], |
| 60 | inputs=[0], |
| 61 | outputs=[1], |
| 62 | chains=[ |
| 63 | Chain( |
| 64 | inputs=[], |
| 65 | outputs=[], |
| 66 | instructions=[Instruction(KernelCall(op_index=0, args=[0, 1]))], |
| 67 | stacktrace=None, |
| 68 | ) |
| 69 | ], |
| 70 | container_meta_type=ContainerMetadata( |
| 71 | encoded_inp_str="place", encoded_out_str="place" |
| 72 | ), |
| 73 | operators=[Operator(name="aten::add", overload="Tensor")], |
| 74 | delegates=[], |
| 75 | non_const_buffer_sizes=[0, 1024], |
| 76 | ) |
| 77 | ], |
| 78 | constant_buffer=[], |
| 79 | backend_delegate_data=[], |
| 80 | segments=[], |
| 81 | constant_segment=SubsegmentOffsets(segment_index=0, offsets=[]), |
| 82 | named_data=[], |
| 83 | ) |
| 84 | |
| 85 | |
| 86 | def register_additional_test_aten_ops() -> None: |