Fix RuntimeError: pidfd_getfd: Operation not permitted

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

Author: kevalmorabia97

modelopt/torch/puzzletron/subblock_stats/calc_subblock_stats.py

@@ -25,8 +25,6 @@
 from pathlib import Path
 from typing import Iterable, Optional, Type, TypeVar
 
-os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
-
 import pandas as pd
 import torch
 from immutabledict import immutabledict