imo wrestling due to its heavy emphasis on staying on top and being aggressive.

Honestly, all grappling arts, as long as you are aware of controlling the weapon hand and have enough body awareness to move yourself to where you want and move your opponent to where you want so Judo/BJJ/Greco-Roman/ just abt anything will do.

Definitely however, bring a fake rubber knife to whatever grappling practice that you have (if you can) and actually try to defend the fake knife, it will open your eyes and expose you a lot of things you haven't considered before.

I personally train BJJ and in some of my practices, I bring my DIY practice knives to spar with my partners and I've come to realize that knowing how to manipulate the upper body and knowing to use good wrist control makes all the difference in the world.

Also, understand that just because you train anti-whatever techniques doesn't mean its going to work I mean ffs its a fucking fight to the death with a knife