Yep.
Although, there's an important detail missing: It may not be illegal (against the law) to possess it, but it is still banned from possession via prison rules. Just like a cell phone or pornography. Not illegal but against the rules.
So you can be prosecuted for breaking the prison penal code.
California court rules inmates can have pot in prison if they don't consume it
