Henson
Henson
Does thundersvm support multi-GPU training?will multi-GPU make the training faster?
I'm interested in your work, I want to know when will the code be released.Look forward to your reply.
latest yolov5 merge into master
``` import torch import torch.nn.functional as F def export_linalg_by_shark(model, dummy_input): from extension.shark.shark_importer import SharkImporter mlir_type = "linalg" mlir_importer = SharkImporter( model, (dummy_input,), frontend="torch", return_str=True, ) mlir_str = mlir_importer._torch_mlir( is_dynamic=False, tracing_required=True,...
``` def export_mlir(model, dummy_input): from torch_mlir import torchscript module = torchscript.compile(model, dummy_input, output_type="tosa") print("TOSA OutputType\n") with open(f"test.mlir", "wt") as f: print(module.operation.get_asm(), file=f) if DEVICE == 'cuda': collate_fn = lambda x:...