torchmatch
constant

transport.samples.kernels.grad_sqeuclid.SHARED_MEM_BUDGET_KB

SHARED_MEM_BUDGET_KB = 96

Source: torchmatch/transport/samples/kernels/grad_sqeuclid.py:883