(self, *args, **kwargs)
| 119 | return pruned_configs |
| 120 | |
| 121 | def warmup(self, *args, **kwargs): |
| 122 | self.nargs = dict(zip(self.arg_names, args)) |
| 123 | for config in self.prune_configs(kwargs): |
| 124 | self.fn.warmup( |
| 125 | *args, |
| 126 | num_warps=config.num_warps, |
| 127 | num_stages=config.num_stages, |
| 128 | **kwargs, |
| 129 | **config.kwargs, |
| 130 | ) |
| 131 | self.nargs = None |
| 132 | |
| 133 | |
| 134 | def autotune(configs, key, prune_configs_by=None, reset_to_zero=None, nearest_power_of_two=False): |
nothing calls this directly
no test coverage detected