Method setup

bench/python/all_ops/op_cache_limit.py:106–111

(self, input)

# This is the {random, large} case. This is the worst case scenario, never re-using the cache
class OpCacheLimitLargeAndRandom(BaseOpCacheLimit):
    def setup(self, input):
        # Set the cache limit to the total gpu memory for this benchmark
        # low=1000, high=2000 results in always creating tensor's of random shape
        # between [(1000,1000,3), (1999,1999,3)]
        total = torch.cuda.mem_get_info()[1]
        super().setup(input, total, low=1000, high=2000)

    def run(self, input):
        super().run(input)