Description
🐛 Describe the bug
Create a large job for MPS to work on and we hit an inbuilt limit that bombs out with...
/AppleInternal/Library/BuildRoots/20d6c351-ee94-11ec-bcaf-7247572f23b4/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShaders/MPSCore/Types/MPSNDArray.mm:705: failed assertion `[MPSTemporaryNDArray initWithDevice:descriptor:] Error: product of dimension sizes > 2**31'
This only occurs on MPS, I can easily run larger jobs on Cuda
Versions
Not overly helpful...
Collecting environment information...
Traceback (most recent call last):
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 492, in
main()
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 475, in main
output = get_pretty_env_info()
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 470, in get_pretty_env_info
return pretty_str(get_env_info())
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 319, in get_env_info
pip_version, pip_list_output = get_pip_packages(run_lambda)
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 301, in get_pip_packages
out = run_with_pip(sys.executable + ' -mpip')
File "/Users/ec2-user/Library/Lartis/collect_env.py", line 289, in run_with_pip
for line in out.splitlines()
AttributeError: 'NoneType' object has no attribute 'splitlines'