Given a demand of higher version pytorch(e.g. 1.11), some modification for the cuda code has proposed. Note that the lower version has not been tested, some errors may be met.
When running the test.py, some warnings aren't clear , direct and confusing, so add the traceback module.