MCPcopy Create free account
hub / github.com/NVIDIA/cutlass / get_conv_problems

Function get_conv_problems

test/python/cutlass/conv2d/conv2d_test_utils.py:393–428  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

391
392
393def get_conv_problems():
394 # 64: minimum channel size
395 conv_problems = TestbedConv2dProblemSizes(64).all
396
397 # Insert alignment 4 & 2 tests
398 conv_problems += [
399 Conv2DProblemSize(
400 1, 4, 4, 12,
401 8, 3, 3, 12,
402 0, 0,
403 3, 3,
404 1, 1,
405 ConvMode.CrossCorrelation,
406 1, 1
407 ),
408 Conv2DProblemSize(
409 1, 4, 4, 14,
410 8, 3, 3, 14,
411 0, 0,
412 3, 3,
413 1, 1,
414 ConvMode.CrossCorrelation,
415 1, 1
416 ),
417 Conv2DProblemSize(
418 1, 23, 56, 98,
419 128, 3, 3, 98,
420 4, 5,
421 3, 3,
422 1, 1,
423 ConvMode.CrossCorrelation,
424 1, 1
425 ),
426 ]
427
428 return conv_problems

Callers 1

conv2d_sm80.pyFile · 0.85

Calls 2

Conv2DProblemSizeClass · 0.90

Tested by

no test coverage detected