fix: Fix _get_accelerator_memory#44

svskvar wants to merge 2 commits into

mainamazon-contributing/upstream-to-pytorch:mainfrom

dtensor_get_memory_info_fixamazon-contributing/upstream-to-pytorch:dtensor_get_memory_info_fixCopy head branch name to clipboard