Artificial Intelligence, Machine Learning, Parallel Computing, Diffusion Models, Content Generation.
computation system uses ColoTracer, a forked version of the torch.FX Tracer, to guide the search for an optimal parallelization strategy. The meta-information of each tensor, such as tensor shape, dims, dtype, etc., is computed and recorded during the tracing process. This approach has the