Add setting of env flag when apex is turned on

ROCm / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

BSD 3-Clause "New" or "Revised" License

17 stars 14 forks source link

Open alugorey opened 1 year ago

alugorey commented 1 year ago

This flag allows exterior entities to query whether apex.amp is turned on at runtime.

alugorey commented 1 year ago

@jithunnair-amd I can't imagine my one line change to set an env var has anything to do with the CI failures. Links don't lead anywhere. Ideas?

alugorey commented 10 months ago

@jeffdaily Can we get another look a this?