The cpu is a Xeon W-3235. According to the specification the processor should have two AVX-512 instruction units.

  
      
      ark.intel.com
  
  
    

Intel® Xeon® W-3235 Processor (19.25M Cache, 3.30 GHz) Product Specifications

Intel® Xeon® W-3235 Processor (19.25M Cache, 3.30 GHz) quick reference guide including specifications, features, pricing, compatibility, design documentation, ordering codes, spec codes and more.


  
  
    
    
  
  


The error is only raised when I choose the dimensions of the image quite high. The following would work, but not the larger values above.
f1 = torch.randn(1,64,2400,3200)

With large values, f.conv2d crashes later with the meaningless error. Any other way to debug this?