cupy.testing.for_dtypes_combination#

cupy.testing.for_dtypes_combination(types, names=('dtype',), full=None)[source]#

一个装饰器,用于检查固定装置(fixture)与 dtype 的乘积集。

参数:
  • types (列表,包含 dtypes) – 要测试的 dtype。

  • names (列表,包含 字符串) – 传入 dtypes 的参数名。

  • full (布尔值) – 如果为 True,则测试所有 dtype 组合。否则,将测试组合的子集(参见下面的描述)。

装饰器将 names 指定的关键字参数添加到测试固定装置中。然后,它并行运行固定装置,并传入 (可能是一个子集) dtype 的乘积集。dtype 的范围由 types 指定。

要测试的 dtype 组合取决于选项 full。如果 fullTrue,则测试 types 的所有组合。有时,这种详尽的测试可能代价高昂。因此,如果 fullFalse,则只随机抽样可能组合的子集。如果 fullNone,则行为由环境变量 CUPY_TEST_FULL_COMBINATION 决定。如果该值设置为 '1',则其行为等同于 full=True,否则等同于 full=False