@@ -972,37 +972,49 @@ def _add_workload_vstack(array_pool):
972
972
973
973
974
974
def _add_workload_equal (array_pool ):
975
- OpArgMngr .add_workload ('equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
975
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
976
+ # OpArgMngr.add_workload('equal', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
977
+ OpArgMngr .add_workload ('equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
976
978
OpArgMngr .add_workload ('equal' , np .array ([np .nan ]), np .array ([np .nan ]))
977
979
OpArgMngr .add_workload ('equal' , array_pool ['4x1' ], array_pool ['1x2' ])
978
980
979
981
980
982
def _add_workload_not_equal (array_pool ):
981
- OpArgMngr .add_workload ('not_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
983
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
984
+ # OpArgMngr.add_workload('not_equal', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
985
+ OpArgMngr .add_workload ('not_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
982
986
OpArgMngr .add_workload ('not_equal' , np .array ([np .nan ]), np .array ([np .nan ]))
983
987
OpArgMngr .add_workload ('not_equal' , array_pool ['4x1' ], array_pool ['1x2' ])
984
988
985
989
986
990
def _add_workload_greater (array_pool ):
987
- OpArgMngr .add_workload ('greater' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
991
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
992
+ # OpArgMngr.add_workload('greater', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
993
+ OpArgMngr .add_workload ('greater' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
988
994
OpArgMngr .add_workload ('greater' , array_pool ['4x1' ], array_pool ['1x2' ])
989
995
OpArgMngr .add_workload ('greater' , np .array ([np .nan ]), np .array ([np .nan ]))
990
996
991
997
992
998
def _add_workload_greater_equal (array_pool ):
993
- OpArgMngr .add_workload ('greater_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
999
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
1000
+ # OpArgMngr.add_workload('greater_equal', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
1001
+ OpArgMngr .add_workload ('greater_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
994
1002
OpArgMngr .add_workload ('greater_equal' , array_pool ['4x1' ], array_pool ['1x2' ])
995
1003
OpArgMngr .add_workload ('greater_equal' , np .array ([np .nan ]), np .array ([np .nan ]))
996
1004
997
1005
998
1006
def _add_workload_less (array_pool ):
999
- OpArgMngr .add_workload ('less' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
1007
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
1008
+ # OpArgMngr.add_workload('less', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
1009
+ OpArgMngr .add_workload ('less' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
1000
1010
OpArgMngr .add_workload ('less' , array_pool ['4x1' ], array_pool ['1x2' ])
1001
1011
OpArgMngr .add_workload ('less' , np .array ([np .nan ]), np .array ([np .nan ]))
1002
1012
1003
1013
1004
1014
def _add_workload_less_equal (array_pool ):
1005
- OpArgMngr .add_workload ('less_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float16 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float16 ))
1015
+ # TODO(junwu): fp16 does not work yet with TVM generated ops
1016
+ # OpArgMngr.add_workload('less_equal', np.array([0, 1, 2, 4, 2], dtype=np.float16), np.array([-2, 5, 1, 4, 3], dtype=np.float16))
1017
+ OpArgMngr .add_workload ('less_equal' , np .array ([0 , 1 , 2 , 4 , 2 ], dtype = np .float32 ), np .array ([- 2 , 5 , 1 , 4 , 3 ], dtype = np .float32 ))
1006
1018
OpArgMngr .add_workload ('less_equal' , array_pool ['4x1' ], array_pool ['1x2' ])
1007
1019
OpArgMngr .add_workload ('less_equal' , np .array ([np .nan ]), np .array ([np .nan ]))
1008
1020
0 commit comments