update

infer_api.py

@@ -99,6 +99,28 @@ weight_dtype = torch.float16
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 VIEWS = ['front', 'front_right', 'right', 'back', 'left', 'front_left']
 
+
+#### TEST ####
+import nvdiffrast.torch as dr
+import torch
+from typing import Tuple
+
+@spaces.GPU
+def _warmup(glctx, device=None):
+    device = 'cuda' if device is None else device
+    #windows workaround for https://github.com/NVlabs/nvdiffrast/issues/59
+    def tensor(*args, **kwargs):
+        return torch.tensor(*args, device=device, **kwargs)
+    pos = tensor([[[-0.8, -0.8, 0, 1], [0.8, -0.8, 0, 1], [-0.8, 0.8, 0, 1]]], dtype=torch.float32)
+    tri = tensor([[0, 1, 2]], dtype=torch.int32)
+    dr.rasterize(glctx, pos, tri, resolution=[256, 256])
+
+_glctx = dr.RasterizeCudaContext(device=None)
+_warmup(_glctx, device)
+
+#### TEST END ####
+
+
 repo_id = "hyz317/StdGEN"
 all_files = list_repo_files(repo_id, revision="main")
 for file in all_files: