| 103 | |
| 104 | |
| 105 | OpSpec SpecMetaDev(const std::string &device) { |
| 106 | return OpSpec("StreamAssignmentMetaOp") |
| 107 | .AddArg("device", device) |
| 108 | .AddArg("num_threads", 1) |
| 109 | .AddArg("max_batch_size", 1); |
| 110 | } |
| 111 | |
| 112 | OpSpec SpecMetaGPU() { |
| 113 | return SpecMetaDev("gpu"); |
no test coverage detected