AttributeError: module 'cutlass_lib.library' has no attribute 'GroupMode'

[kisisjrlly]

when I run examples, an error An error occurred： File "/opt/conda/lib/python3.8/site-packages/aitemplate/backend/cuda/conv2d/conv2d_bias_relu_few_channels.py", line 47, in conv2d_bias_relu_few_channels_config func_attrs["op_instance"] = cfc.extract_config( File "/opt/conda/lib/python3.8/site-packages/aitemplate/backend/cuda/conv2d/common_conv2d_few_channels.py", line 49, in extract_config return common.extract_config( File "/opt/conda/lib/python3.8/site-packages/aitemplate/backend/cuda/conv2d/common.py", line 567, in extract_config ret = f_proc_op(op) File "/opt/conda/lib/python3.8/site-packages/aitemplate/backend/cuda/conv2d/common.py", line 548, in f_proc_op op = f_apply_special_config(func_attrs, op) File "/opt/conda/lib/python3.8/site-packages/aitemplate/backend/cuda/conv2d/common_conv2d_few_channels.py", line 34, in apply_special_config op.group_mode = cutlass_lib.library.GroupMode.NoneGroup

command i run: git clone --recursive https://github.com/facebookincubator/AITemplate cd AITemplate/python python setup.py bdist_wheel python dist/aitemplate-0.1.dev1-py3-none-any.whl cd ../examples/01_resnet-50/ python benchmark_ait.py

[ipiszy]

Hi @kisisjrlly , what's your GPU arch? You'll need SM80 (or at least SM70) to run AITemplate. Thanks.