xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/loopnest_randomization.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 namespace torch::jit::tensorexpr {
4 
5 // Applies a series of loop optimizations chosen randomly. This is only for
6 // testing purposes. This allows automatic stress testing of NNC loop
7 // transformations.
8 void loopnestRandomization(int64_t seed, LoopNest& l);
9 } // namespace torch::jit::tensorexpr
10