Remove extra bracket, update header definition (#86317)
Summary: Fix compilation error
Test Plan: Unit test
Reviewed By: malfet, mikaylagawarecki
Differential Revision: D40108369
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86317
Approved by: https://github.com/malfet
