This is the MATLAB code implementation for the WMMSE algorithm.
CLick here for the original paper link: An Iteratively Weighted MMSE Approach to Distributed Sum-Utility Maximization for a MIMO Interfering Broadcast Channel

WMMSE.m : The main function.
find_U.m : The function for finding the U in each iteration of the WMMSE algorithm.
find_W.m : The function for finding the W in each iteration of the WMMSE algorithm.
find_V.m : The function for finding the V in each iteration of the WMMSE algorithm.
sum_rate.m : The function for computing the weighted sum rate.

Run WMMSE.m in matlab and get the following figure: