(self)
| 107 | }) |
| 108 | |
| 109 | def benchmark_split(self): |
| 110 | print("Forward vs backward concat") |
| 111 | shapes = [[2000, 8], [8, 2000], [100, 18], [1000, 18], [10000, 18], |
| 112 | [100, 97], [1000, 97], [10000, 1], [1, 10000]] |
| 113 | axis_ = [1] # 0 is very fast because it doesn't actually do any copying |
| 114 | num_outputs = 100 |
| 115 | variable = [False, True] # fixed input size or not |
| 116 | for shape in shapes: |
| 117 | for axis in axis_: |
| 118 | for v in variable: |
| 119 | self._run_graph("gpu", shape, v, num_outputs, axis) |
| 120 | |
| 121 | |
| 122 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected