AMD MI210 ResNet test error

[crispyberry]

Hi, we I use HIP_VISIBLE_DEVICES=0 python3 benchmark_ait.py, I meet errors like this:

.../aitemplate/compiler/ops/conv/conv2d.py", line 427, in gen_profiler
func(self._attrs, dtype=self._attrs["inputs"][0]._attrs["dtype"])
TypeError: conv2d_config() got an unexpected keyword argument 'dtype'

[ipiszy]

Seems that ROCM backend doesn't have dtype support. You could try to modify

AITemplate/python/aitemplate/backend/rocm/conv2d/conv2d.py

Line 25 in 6bbe03c

def conv2d_config(func_attrs):

to add dtype param, following

AITemplate/python/aitemplate/backend/cuda/conv2d/conv2d.py

Line 25 in 6bbe03c

def conv2d_config(

.

[fsx950223]

Try to use the branch https://github.com/ROCmSoftwarePlatform/AITemplate/tree/merge_upstream.

[crispyberry]

Thanks!